torch
torchvision
torchtyping
einops
diffusers
transformers
accelerate
xformers
wandb
fastprogress
matplotlib