torch>=1.10.0
numpy>=1.20.0

[dev]
pytest>=7.0
pytest-cov
black
isort
mypy
