torch
datasets
sentence-transformers
transformers