numpy>=1.17
packaging>=20.0
psutil
pyyaml
torch>=1.4.0

[dev]
black~=23.1
ruff>=0.0.241
hf-doc-builder>=0.3.0
pytest
pytest-xdist
pytest-subtests
parameterized
datasets
evaluate
transformers
scipy
scikit-learn
deepspeed
tqdm
rich

[docs]

[quality]
black~=23.1
ruff>=0.0.241
hf-doc-builder>=0.3.0

[rich]
rich

[sagemaker]
sagemaker

[test_dev]
datasets
evaluate
transformers
scipy
scikit-learn
deepspeed
tqdm

[test_prod]
pytest
pytest-xdist
pytest-subtests
parameterized

[test_trackers]
wandb
comet-ml
tensorboard

[testing]
pytest
pytest-xdist
pytest-subtests
parameterized
datasets
evaluate
transformers
scipy
scikit-learn
deepspeed
tqdm
