diffrax has 252K monthly downloads and 2K GitHub stars. Version 0.7.2 supports Python 3.11+, first published in 2022.
It is actively maintained under a permissive license, last released in Feb 2026.
It is built on top of equinox, jax and jaxtyping. It is used by 105 tracked packages, including dynamiqs, openscvx and flowmc.
GitHub topics include differential-equations, jax and deep-learning.
Package Insights
- No releases in 123 days despite active commits
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- equinox >=0.11.10
- jax >=0.4.38
- jaxtyping >=0.2.24
- lineax >=0.0.5
- optimistix >=0.1.0
- typing-extensions >=4.5.0
- wadler-lindig >=0.1.1
16 optional dependencies
- beartype[tests]
- griffe[docs]
- hippogriffe[docs]
- jaxlib[tests]
- mkdocs[docs]
- mkdocs-include-exclude-files[docs]
- mkdocs-ipynb[docs]
- mkdocs-material[docs]
- mkdocstrings[docs]
- mkdocstrings-python[docs]
- optax[tests]
- pre-commit[dev]
- pymdown-extensions[docs]
- pytest[tests]
- scipy[tests]
- tqdm[tests]
Used By
- dynamiqs
- openscvx
- flowmc
- desc-opt
- sbijax
- fwl-zalmoxis
- tvboptim
- circulax
- spectraxgk
- pyrcel
- qutip-jax
- epimodels
- moscot
- bayinx
- cd-dynamax
- abcmb
- trajopt
- astrodynx
- qml-essentials
- pycollimator
- jaxace
- climemu
- bijx
- jax-sysid
- jaxley-mech
- desc-v2
- pyrovelocity
- clearbox-synthetic-kit
- dynestyx
- rectified-flows
- cellflow-tools
- off
- diffraxtra
- sbgm
- catalax
- sigway
- raytrax
- sbiax
- jaxprop
- eulerpi
- safe-autonomy-simulation
- openreservoircomputing
- gensbi
- nurbspy
- jaxkineticmodel
- pcgym
- opifex
- hyprfine
- essos
- stormi