GitHub repository: vishalbelsare/torchcde
Primary language: Python
Description: Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.