Improve `NeuralSDE` class to include `NeuralODE` features

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

Improve `NeuralSDE` class to include `NeuralODE` features #93

Open Zymrael opened 3 years ago

Zymrael commented 3 years ago

Standardize the two classes. NeuralSDEs should have an option to use our internals or torchsde as a backend. Related to this for implementation of our internal solver SDE suite.