torch
numpy
einops
huggingface_hub
