Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation
This library provides ordinary differential equation (ODE) solvers implemented in PyTorch. Backpropagation through ODE solutions is supported using the adjoint method for constant memory cost. For usage of ODE solvers in deep learning applications. As the solvers are implemented in PyTorch, algorithms in this repository are fully supported to run on the GPU.
Release | Stable | Testing |
---|---|---|
Fedora Rawhide | 0.2.5-3.fc42 | - |
Fedora 42 | 0.2.5-3.fc42 | - |
You can contact the maintainers of this package via email at
python-torchdiffeq dash maintainers at fedoraproject dot org
.