diffrax (pypi): GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.
Keywords: deep-learning|differential-equations|dynamical-systems|equinox|jax|machine-learning|neural-differential-equations|neural-networks