torch>=2.0
torchao>=0.4
numpy
pyyaml

[all]
torch-pruning
bitsandbytes
torchprofile

[dev]
pytest>=7.0
pytest-mock

[test-all]
transformers
openai-whisper
ultralytics
timm
