GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.
No known vulnerabilities in the latest version
Based on latest version 0.7.2. If you're running an older version, check OSV for your specific version.
[](/packages/diffrax)
<a href="/packages/diffrax"><img src="/api/badges/diffrax?period=month" alt="PyPI Stats"></a>