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