accelerate
datasets
faiss-gpu
transformers
wandb