patrick-kidger/diffrax

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

[view on github]last commit: Sep 6, 2026

stars

2,115

7d

+3

30d

+10

90d

+52

## star history

## found in

Awesome Open Source AI/Deep Learning Frameworks