numpy
wandb
torch
seaborn
Pillow
torchvision
diffusers
s3fs
scipy
webdataset
datasets
imagecorruptions