einops
imageio
matplotlib
numpy
pandas
scipy
seaborn
timm
torch
torchvision
tqdm
vector-quantize-pytorch
lightning