torch
transformers
sentence-transformers
scikit-learn
matplotlib
numpy