torch
torchvision
transformers
kornia
einops
timm