torchdyn icon indicating copy to clipboard operation
torchdyn copied to clipboard

Improve `NeuralSDE` class to include `NeuralODE` features

Open Zymrael opened this issue 4 years ago • 0 comments

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.

Zymrael avatar Jul 20 '21 09:07 Zymrael