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

How to do backpropagation with multiple shooting? #213

Open TJHeeringa opened 3 months ago

TJHeeringa commented 3 months 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?