v 0.2.5 Updated: 3 weeks, 6 days ago
Differentiable ODE solvers with full GPU support and adjoint backpropagation
torchdiffeq provides differentiable ODE solvers with full GPU support and O(1)-memory backpropagation using the adjoint method.
https://github.com/rtqichen/torchdiffeq
| Installations | 0 |
| Requested Installations | 0 |