numpy>=1.17
packaging>=20.0
psutil
pyyaml
torch>=1.13.0
transformers
accelerate

[dev]
black~=22.0
ruff>=0.0.241
hf-doc-builder

[docs_specific]
hf-doc-builder

[quality]
black~=22.0
ruff>=0.0.241

[test]
black~=22.0
ruff>=0.0.241
hf-doc-builder
pytest
pytest-xdist
parameterized
datasets