torch
torchvision
torchaudio
transformers
accelerate
Pillow
gradio
einops