torch>=1.9.0
numpy>=1.21.0

[dev]
pytest>=6.0
black
flake8
mypy
