SDE solvers and stochastic adjoint sensitivity analysis in PyTorch.
conda install conda-forge::torchsde