torch>=1.9.0
numpy>=1.19.0

[dev]
pytest
black
