datasets
transformers
torch
pandas
faiss-cpu
einops