numpy
pytorch
transformers
sentence-transformers
sklearn