transformers
torch
torchvision
timm
numpy
diffusers