torch
torchvision
kornia
transformers
pillow
matplotlib
timm