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

`torchdyn` backend in `torchcde` #94

Closed Zymrael closed 2 years ago

Zymrael commented 3 years ago

Introduce torchdyn as a backend for torchcde. This would make it easier to further experiment with Neural CDEs + multiple shooting methods, and would streamline our prototypes so that the tutorials can be easier. Pretrained Neural CDEs will then be added to our model zoo.

@patrick-kidger pinging so that you can keep track of progress.