torch
transformers
numpy
scipy
cupy-cuda12x
tqdm
datasets
