torch
torchvision
transformers
einops
pillow
gradio
accelerate