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

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

[notebooks]
ipywidgets

[test]
pytest-skip-slow
pytest
mock
