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 2, 2026 at 10:24
License
Apache-2.0
Total Downloads
45.6K
Version Downloads
2.4K
Supported Platforms
GitHub Repository
https://github.com/patrick-kidger/diffraxDocumentation
https://docs.kidger.site/diffrax/