torch>=2.0.0
huggingface_hub>=0.20.0

[dev]
pytest
build
twine

[gpu]
triton
cuda-toolkit
