torch
numpy
einops
huggingface_hub

[dev]
