Implementations of differentiable ODE and PDE solvers for gradient based initial state optimiziation in JAX.
Primary LanguageJupyter Notebook