diffrax
Neural Differential Equations
Numerical differential equation solving in JAX
- Repository
- github.com/patrick-kidger/diffrax
Source attribution
- Awesome AI for Science — github.com/patrick-kidger/diffrax
Related resources
Julia differential equations suite
PyTorch implementation of neural ODEs
Neural differential equations in PyTorch
Neural differential equations in Julia