GitHub repository: FedericoV/diffrax
Description: Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.