numpy
scipy
torch
Pillow

[all]
wandb
seaborn
torchvision
diffusers
s3fs
webdataset
datasets
imagecorruptions

[dev]
pytest
black
flake8

[docs]
sphinx
sphinx-rtd-theme
