
[deepspeed]
deepspeed

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

[docs]

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

[rich]
rich

[sagemaker]
sagemaker

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

[test_fp8]
torchao

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

[test_trackers]
comet-ml
dvclive
matplotlib
mlflow
swanlab
tensorboard
trackio
wandb

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