sentence-transformers
datasets
torch