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