torch>1.0.0
numpy

[dev]
black==22.6.0
pre-commit>=2.18
pytest==7.1.1
pytest-cov==3.0.0
