GitHub repository: patrick-kidger/torchsde
Description: Differentiable SDE solvers with GPU support and efficient sensitivity analysis.