GitHub repository: pandapth/neurode_from_torchdiffeq
Primary language: Python
Description: Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.