Currently, this repo does not implement SDE solvers; this could be done e.g. using [this package](https://github.com/google-research/torchsde)