torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
5k stars
125 watching
909 forks
Language: Python
last commit: 12 months ago torchdiffeq Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.