torchsde：具有GPU支持和高效灵敏度分析的可区分SDE求解器

差分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 , 2 t_size = 20 class SDE ( torch . nn . Module ): no