accelerate
beartype
einops>=0.6
lion-pytorch
torch>=1.6
tqdm
