numpy>=1.17
packaging>=20.0
psutil
pyyaml
torch>=2.0.0
huggingface_hub>=0.21.0
safetensors>=0.4.3

[deepspeed]
deepspeed

[dev]
ruff==0.13.1
pytest>=7.2.0
pytest-xdist
pytest-subtests
parameterized
pytest-order
datasets
diffusers
evaluate
torchdata>=0.8.0
torchpippy>=0.2.0
transformers
scipy
scikit-learn
tqdm
bitsandbytes
timm
rich

[docs]

[quality]
ruff==0.13.1

[rich]
rich

[sagemaker]
sagemaker

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

[test_fp8]
torchao

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

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

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