torchmetrics
scikit-learn
backports.strenum
python-dotenv
GitPython
matplotlib
faiss-cpu
pandas
fire
lightning

[data]
transformers
datasets
torchvision

[dev]
black
flake8
isort
pre-commit
bandit
mkdocs
mkdocs-material
mike
pytest
pytest-cov
transformers
datasets
torchvision

[docs]
mkdocs
mkdocs-material
mike

[test]
pytest
pytest-cov
