torch
torchvision
sentence_transformers
matplotlib