pytorch_lightning
deepspeed
taming-transformers
scipy
einops
kornia
open_clip_torch
openai-clip
xformers
timm
av
gradio