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