py-torchdiffeq

v 0.2.5 Updated: 3 weeks, 6 days ago

Differentiable ODE solvers with full GPU support and adjoint backpropagation

torchdiffeq provides differentiable ODE solvers with full GPU support and O(1)-memory backpropagation using the adjoint method.

torchdiffeq provides differentiable ODE solvers with full GPU support and O(1)-memory backpropagation using the adjoint method.

+ Show more

https://github.com/rtqichen/torchdiffeq

Installable ports: