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.4k stars 130 forks source link

Closed fedebotu closed 2 years ago