torch>=2.0
numpy
tqdm
matplotlib
scipy
