torch
torchvision
transformers
timm
gradio