beartype
einops
lion-pytorch
torch
torchvision
tqdm
accelerate>=0.25.0
wandb
