transformers
torch
bitsandbytes
flamingo_pytorch
pillow
accelerate
datasets
rich
tensorboard
wandb
boto3
lion-pytorch
fairscale
