torch
transformers
pytorch-lightning
datasets
sentence-transformers
numpy