transformers==4.17.0
wandb
datasets==1.18.0
sentencepiece==0.1.96
ujson
pytrec-eval
scipy