Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ci: add NNODE, NeuralAdapter, IntegroDiff in GHA from buildkite to have them in Downgrade CI #845

Merged
merged 3 commits into from
Apr 6, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 0 additions & 25 deletions .buildkite/pipeline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,31 +16,6 @@ steps:
# Don't run Buildkite if the commit message includes the text [skip tests]
if: build.message !~ /\[skip tests\]/

- label: "{{matrix.group}} v{{matrix.version}}"
matrix:
setup:
version:
- "1"
group:
- "NNODE"
- "NeuralAdapter"
- "IntegroDiff"
env:
BUILDKITE_PLUGIN_JULIA_VERSION: "{{matrix.version}}"
GROUP: "{{matrix.group}}"
plugins:
- JuliaCI/julia#v1
- JuliaCI/julia-test#v1:
coverage: false
julia_args: "--threads=auto"
agents:
os: "linux"
queue: "juliaecosystem"
arch: "x86_64"
timeout_in_minutes: 680
# Don't run Buildkite if the commit message includes the text [skip tests]
if: build.message !~ /\[skip tests\]/

- label: "Documentation"
plugins:
- JuliaCI/julia#v1:
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,9 @@ jobs:
- Logging
- Forward
- DGM
- NNODE
- NeuralAdapter
- IntegroDiff
version:
- "1"
steps:
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/Downgrade.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,9 @@ jobs:
- Logging
- Forward
- DGM
- NNODE
- NeuralAdapter
- IntegroDiff
version:
- "1"
steps:
Expand Down
10 changes: 5 additions & 5 deletions Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
Adapt = "4"
AdvancedHMC = "0.6.1"
Aqua = "0.8"
ArrayInterface = "7.8"
ArrayInterface = "7.9"
CUDA = "5.2"
ChainRulesCore = "1.21"
ComponentArrays = "0.15.8"
Expand All @@ -55,15 +55,15 @@ DomainSets = "0.6, 0.7"
Flux = "0.14.11"
ForwardDiff = "0.10.36"
Functors = "0.4.4"
Integrals = "4"
Integrals = "4.4"
LineSearches = "7.2"
LinearAlgebra = "1"
LogDensityProblems = "2"
Lux = "0.5.22"
LuxCUDA = "0.3.2"
MCMCChains = "6"
MethodOfLines = "0.11"
ModelingToolkit = "9.7"
ModelingToolkit = "9.9"
MonteCarloMeasurements = "1.1"
Optim = "1.7.8"
Optimization = "3.24"
Expand All @@ -78,8 +78,8 @@ RuntimeGeneratedFunctions = "0.5.12"
SafeTestsets = "0.1"
SciMLBase = "2.28"
Statistics = "1.10"
SymbolicUtils = "1.4"
Symbolics = "5.24"
SymbolicUtils = "1.5"
Symbolics = "5.27.1"
Test = "1"
UnPack = "1"
Zygote = "0.6.69"
Expand Down
4 changes: 3 additions & 1 deletion docs/src/examples/complex.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Complex Equations with PINNs

NeuralPDE supports training PINNs with complex differential equations. This example will demonstrate how to use it for [`NNODE`](@ref). Let us consider a system of [bloch equations](https://en.wikipedia.org/wiki/Bloch_equations). Note [`QuadratureTraining`](@ref) cannot be used with complex equations due to current limitations of computing quadratures.
NeuralPDE supports training PINNs with complex differential equations. This example will demonstrate how to use it for [`NNODE`](@ref). Let us consider a system of [bloch equations](https://en.wikipedia.org/wiki/Bloch_equations) [^1]. Note [`QuadratureTraining`](@ref) cannot be used with complex equations due to current limitations of computing quadratures.

As the input to this neural network is time which is real, we need to initialize the parameters of the neural network with complex values for it to output and train with complex values.

Expand Down Expand Up @@ -95,3 +95,5 @@ plot!(ground_truth.t, imag.(reduce(hcat, ground_truth.u)[4, :]))
```

We can see it is able to learn the real parts of `u1`, `u2` and imaginary parts of `u3`, `u4`.

[^1]: https://steck.us/alkalidata/
Loading