torch
torchvision
pillow
matplotlib
tensorboard
tensorboardX
lmdb
matplotlib
scipy
ninja
einops
einops_exts
braceexpand
transformers
open-clip-torch