numpy
torch
einops 
timm 
torchvision 
functorch  
torch