← All repos

patrick-kidger/diffrax

View on GitHub ↗ ★ 2,122 Python

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

8 mentions on Hacker News by 3 people: 1 story, 7 comments · first 2022-04-14 · latest 2025-01-11