GitHub repository: sarvex/torchsde
Primary language: Python
Description: Differentiable SDE solvers with GPU support and efficient sensitivity analysis.