torch>=2.7
numpy
pyarrow
huggingface_hub
safetensors
typing_extensions
