torch<3,>=2.1
scikit-learn<1.7,>=1.2.0
typing_extensions>=4.4.0
scipy<2,>=1.11.1
pandas<3,>=1.4.0
einops<0.9,>=0.2.0
huggingface-hub<1,>=0.0.1

[dev]
pre-commit
ruff
mypy
pytest
onnx
psutil
mkdocs
mkdocs-material
mkdocs-autorefs
mkdocs-gen-files
mkdocs-literate-nav
mkdocs-glightbox
mkdocstrings[python]
markdown-exec[ansi]
mike
black
