accelerate
datasets
plotly>=5.13.1
scikit-learn
zstandard
torch>=1.13.0
transformers
huggingface_hub>=0.12.1

[dev]
pre-commit
bump2version
pytest-skip-slow
pytest

[docs]
furo
myst-parser
sphinx
sphinx-autodoc-typehints
sphinx-rtd-theme

[notebooks]
ipywidgets

[test]
pytest-skip-slow
pytest-cov
pytest
mock
