torch>=1.10.0
torchmetrics>=0.6.0
numpy
tqdm
typeguard<5.0,>=3.0
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.81.0,>=6.41.0
pre-commit>=2.19
