torch
torchvision
transformers
gradio
pillow
einops
timm