Numerical differential equation solvers in JAX
Data driven modeling and automated discovery of dynamical systems
Fast ODE Solver for Diffusion Probabilistic Model Sampling
Differentiable SDE solvers with GPU support and efficient sensitivity
A multi-modeling and simulation environment to study complex systems