torchsde (pypi): SDE solvers and stochastic adjoint sensitivity analysis in PyTorch.
Keywords: deep-learning|deep-neural-networks|differential-equations|dynamical-systems|neural-differential-equations|pytorch|stochastic-differential-equations|stochastic-processes|stochastic-volatility-models