numpy
torch>=0.4.1

[testing]
pytest
scipy
