torch>=1.10.0
torchmetrics>=0.6.0
numpy
tqdm
beartype
icontract
scikit-learn
scipy

[docs]
mkdocs>=1.3
mkdocs-material
mkdocstrings-python
mkdocs-jupyter
mkdocs-literate-nav
mkdocs-gen-files
mkdocs-section-index

[tests]
pytest>=3.8
hypothesis<6.99.0,>=6.41.0
pre-commit>=2.19
codespell
