torch>=2.1
numpy>=1.24
safetensors>=0.4
huggingface_hub>=0.24

[visualization]
matplotlib>=3.7
