numpy<2.0.0,>=1.17
packaging>=20.0
psutil
pyyaml
torch>=1.10.0
huggingface_hub>=0.21.0
safetensors>=0.3.1

[deepspeed]
deepspeed<=0.14.0

[dev]
black~=23.1
hf-doc-builder>=0.3.0
ruff~=0.2.1
pytest<=8.0.0,>=7.2.0
pytest-xdist
pytest-subtests
parameterized
datasets
diffusers
evaluate
torchpippy>=0.2.0
transformers
scipy
scikit-learn
tqdm
bitsandbytes
timm
rich

[docs]

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

[rich]
rich

[sagemaker]
sagemaker

[test_dev]
datasets
diffusers
evaluate
torchpippy>=0.2.0
transformers
scipy
scikit-learn
tqdm
bitsandbytes
timm

[test_prod]
pytest<=8.0.0,>=7.2.0
pytest-xdist
pytest-subtests
parameterized

[test_trackers]
wandb
comet-ml
tensorboard
dvclive

[testing]
pytest<=8.0.0,>=7.2.0
pytest-xdist
pytest-subtests
parameterized
datasets
diffusers
evaluate
torchpippy>=0.2.0
transformers
scipy
scikit-learn
tqdm
bitsandbytes
timm
