GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.
No known vulnerabilities in the latest version
[](/packages/diffrax)
<a href="/packages/diffrax"><img src="/api/badges/diffrax?period=month" alt="PyPI Stats"></a>