Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.
Primary LanguagePythonApache License 2.0Apache-2.0