/torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

Primary LanguagePythonApache License 2.0Apache-2.0

Watchers