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

WWW: https://github.com/rtqichen/torchdiffeq
