diffrax (conda): Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.
Keywords: deep-learning|differential-equations|dynamical-systems|jax|machine-learning|neural-differential-equations|neural-networks