torch>=2.1
scikit-learn>=1
typing_extensions
scipy
pandas
einops
huggingface-hub

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