numpy>=1.20.0

[dev]
pytest>=6.0
pytest-cov
black
flake8

[torch]
torch>=1.9.0
