Avoid saving autograd func

DiffEqML / torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

https://torchdyn.org

Apache License 2.0

1.35k stars 125 forks source link

Closed fedebotu closed 2 years ago