datasets
accelerate
transformers
torchvision
wandb
peft
webdataset