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