matplotlib
pytorch-lightning>=0.8.4
scikit-learn
torch>=1.6.0
torchcde>=0.2.3
torchsde>=0.2.5
