numpy
torch
transformers
datasets
hnswlib
scikit-learn