torch>=2.4.0
ninja
numpy
safetensors
huggingface_hub

[dev]
pytest
ruff
black

[examples]
torchvision
transformers
diffusers
datasets
accelerate
sentencepiece
scipy
