torch
torchvision
einops
numpy
timm
