torch
torchaudio
torchvision
lightning
wandb
gradio
einops
timm
tqdm
IPython