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

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

[notebooks]
ipywidgets

[test]
pytest-skip-slow
pytest
mock
