tqdm
ipywidgets
matplotlib
hf-transfer
huggingface_hub
torch
numpy==1.24.4
scipy
einops
einx
torch-summary
