einops
timm
torch
torchvision
sentence_transformers
matplotlib