tqdm
matplotlib
torch
einops
torch-summary
hf-transfer
huggingface_hub
ipywidgets
