numpy
pandas
torch
sentence_transformers