Skip to content

[WIP] Add tutorials on UDEs with neural network rates#1489

Open
TorkelE wants to merge 14 commits into
masterfrom
add_ude_tutorial
Open

[WIP] Add tutorials on UDEs with neural network rates#1489
TorkelE wants to merge 14 commits into
masterfrom
add_ude_tutorial

Conversation

@TorkelE

@TorkelE TorkelE commented Jun 15, 2026

Copy link
Copy Markdown
Member

Merging will likely require #1488 to be finished first (which in turn is waiting for the next OrdinaryDiffEq release).

@TorkelE TorkelE closed this Jun 24, 2026
@TorkelE TorkelE reopened this Jun 24, 2026

@sebapersson sebapersson left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for adding, I think it looks good! Left a few minor comments

Lux.Dense(3 => 1, Lux.softplus, use_bias = false),
)
@SymbolicNeuralNetwork U, θ = nn_arch
A(z) = U(z, θ)[1]

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe use U([z, ]) syntax, makes it easier to understand how to have more than a single input to the neural network

```
Here, `U` represents the neural network structure, and `θ` represents its parameters. Additional options for using the `@SymbolicNeuralNetwork` macro are described [here](https://sciml.github.io/ModelingToolkitNeuralNets.jl/stable/api/#ModelingToolkitNeuralNets.@SymbolicNeuralNetwork) and [here](https://sciml.github.io/ModelingToolkitNeuralNets.jl/stable/symbolic_ude_tutorial/). Next, using the following syntax, we can [interpolate](@ref dsl_advanced_options_symbolics_and_DSL_interpolation) this function approximator as a Catalyst model rate.
```@example ude_rate_based
A(z) = U(z, θ)[1]

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

See comment above

Comment thread docs/src/inverse_problems/udes.md Outdated
Comment thread docs/src/inverse_problems/udes.md Outdated
@TorkelE TorkelE closed this Jul 13, 2026
@TorkelE TorkelE reopened this Jul 13, 2026
@TorkelE TorkelE closed this Jul 13, 2026
@TorkelE TorkelE reopened this Jul 13, 2026
@TorkelE TorkelE closed this Jul 13, 2026
@TorkelE TorkelE reopened this Jul 13, 2026
@TorkelE TorkelE changed the title Add tutorials on UDEs with neural network rates [WIP] Add tutorials on UDEs with neural network rates Jul 17, 2026
@TorkelE

TorkelE commented Jul 17, 2026

Copy link
Copy Markdown
Member Author

This one needs a bit of tuning still

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants