← all repositories

google-research/torchsde

PyTorch library providing GPU-accelerated stochastic differential equation solvers with efficient gradient backpropagation.

1.7k stars Python ML Frameworks
torchsde
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This library implements differentiable SDE solvers that support GPU computation and automatic differentiation for training neural stochastic differential equation models. It provides the sdeint function to numerically integrate SDEs with customizable drift and diffusion functions, enabling applications like learning latent stochastic processes from data. The library is part of the neural differential equations research area within deep learning.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.