torch>=1.10.0
torchmetrics>=0.11.0
numpy>=1.20.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
ruff>=0.1.0
mypy>=1.0.0
pre-commit>=3.0.0
interrogate>=1.5.0
types-psutil>=7.0.0
sphinx>=5.0.0
sphinx-rtd-theme>=1.0.0
sphinx-autodoc-typehints>=1.19.0
myst-parser>=1.0.0
sphinx-autobuild>=2021.3.14

[docs]
sphinx>=5.0.0
sphinx-rtd-theme>=1.0.0
sphinx-autodoc-typehints>=1.19.0
myst-parser>=1.0.0
sphinx-autobuild>=2021.3.14
