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.33k stars 124 forks source link

Create coarse/fine solvers in MultipleShooting #187

Closed jcallaham closed 1 year ago

jcallaham commented 1 year ago

Proposal to fix #186. I don't know any context for why this might not work, but it's not giving me any errors so I'll just see what you all think.

Also added a couple of small unit tests to match the quickstart notebook.