diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
1k stars
20 watching
124 forks
Language: Python
last commit: 15 days ago
Linked from 1 awesome list
deep-learningdifferential-equationsdynamical-systemsequinoxjaxmachine-learningneural-differential-equationsneural-networks
n2cholas/awesome-jax