torch>=2.0
huggingface_hub
h5py
numpy