transformers
datasets
faiss-gpu