GitHub repository: m4ta1l/torchdiffeq
Description: Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.