beartype
einops
lion-pytorch
torch
torchvision
tqdm
accelerate
wandb
