numpy
pytest
scipy
torch>=1.8.0
torchdiffeq==0.2.2
torchtyping==0.1.4
