torch>=2.0
numpy<2
safetensors
pyyaml

[dev]
pytest
ruff

[distill]
transformers

[export]
onnx
onnxruntime
onnxscript
