/torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

Primary LanguagePythonMIT LicenseMIT

Stargazers