torch>=1.10
numpy
scipy

[dev]
pytest
matplotlib
