
[dev]
pytest>=6.0
black
flake8

[docs]
sphinx
sphinx-rtd-theme

[gpu]
torch>=1.9.0
