numpy
torch
transformers
sentence-transformers
sklearn