tqdm
ipywidgets
huggingface_hub
torch
einops
einx
torch-summary
matplotlib
networkx
scipy
scikit-learn
numpy
