torch>=2.0.0
numpy<2.0.0
tqdm
matplotlib
