numpy
pytest
scipy
torch>=1.8.0
torchdiffeq==0.2.1
