search

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.36k stars 125 forks source link

How to do backpropagation with multiple shooting? #213

Open TJHeeringa opened 4 weeks ago

TJHeeringa commented 4 weeks ago

In tutorials m2b and m4c the forward pass of the multiple shooting is shown. However, the example doesn't show how to do backpropagation for those examples. Can you show how to expand these examples with backpropagation?