差分SDE解算器的PyTorch实现 该库提供具有GPU支持和高效反向传播的求解器。 安装 pip install git+https://github.com/google-research/torchsde.git 要求： Python> = 3.6和PyTorch> = 1.6.0。 文献资料 可用。 例子 快速示例 import torch import torchsde batch_size , state_size , brownian_size = 32 , 3 ,