numpy
scipy
torch>=1.7.1

[testing]
pytest
