torch>=2.6.0
ninja
numpy
safetensors
huggingface_hub

[dev]
pytest
ruff

[examples]
torchvision
transformers
diffusers
datasets
accelerate
sentencepiece
scipy
