torch
transformers
scipy
tqdm
