GitHub repository: packquickly/diffrax
Primary language: Python
Description: Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/