torch>=1.9.0
numpy>=1.19.0

[all]
pytest>=6.0
pytest-cov>=2.0
black>=22.0
isort>=5.0
flake8>=4.0
psutil>=5.8.0
sphinx>=4.0
sphinx-rtd-theme>=1.0

[benchmarks]
psutil>=5.8.0

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=22.0
isort>=5.0
flake8>=4.0

[docs]
sphinx>=4.0
sphinx-rtd-theme>=1.0
