DiffEqML / diffeqml-research

This repository contains code released by DiffEqML Research
Apache License 2.0
84 stars 8 forks source link

Differentiable Multiple Shooting Layers Code #5

Open QiyaoWei opened 2 years ago

QiyaoWei commented 2 years ago

Dear authors,

Hope this message finds you well. I am writing to ask whether the code for the MSL paper will be uploaded here soon? In addition, Appendix C of the paper mentions that there will be a pytorch library (presumably accompanying the paper). Is that library the same as this one? Thanks a lot!

Zymrael commented 2 years ago

Hi @QiyaoWei! You can find MultipleShootingLayers in torchdyn (see here and here). We have all the code for the experiments ready but were waiting for functorch to be available in the stable version of PyTorch. The code in its original form will be added here soon, but I'd recommend also checking out the improved version in torchdyn once that comes out.

ddrous commented 1 year ago

@Zymrael Dear authors, Just checking if thing have improved on the PyTorch side, or if you have reconsidered uploading the code in its original form. Many thanks.