diffrax
Numerical differential equation solvers in JAX.
Numerical differential equation solvers in JAX.
To install this package, run one of the following:
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.
Summary
Numerical differential equation solvers in JAX.
Last Updated
Feb 22, 2026 at 13:22
License
Apache-2.0
Supported Platforms
GitHub Repository
https://github.com/patrick-kidger/diffraxDocumentation
https://docs.kidger.site/diffrax/