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.

Sign up for GitHub

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

Jump to bottom

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

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

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
1c110c5
ci: add NNODE, NeuralAdapter, IntegroDiff in GHA from buildkite to ha…
sathvikbhagavan Apr 4, 2024
4fe0529
build: bump versions to get QA passing in Downgrade
sathvikbhagavan Apr 4, 2024
41fa434
docs: add reference for the bloch equations in the complex example
sathvikbhagavan Apr 4, 2024
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
View file
Open in desktop
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
View file
Open in desktop
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
View file
Open in desktop
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
View file
Open in desktop
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
View file
Open in desktop
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