Skip to content

Commit

Permalink
Merge pull request #845 from SciML/sb/ci
Browse files Browse the repository at this point in the history
ci: add NNODE, NeuralAdapter, IntegroDiff in GHA from buildkite to have them in Downgrade CI
  • Loading branch information
ChrisRackauckas authored Apr 6, 2024
2 parents 422434d + 41fa434 commit 49efc20
Show file tree
Hide file tree
Showing 5 changed files with 14 additions and 31 deletions.
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/

0 comments on commit 49efc20

Please sign in to comment.