torch>=2.5.1
tqdm>=4.67.1
numpy>=2.0.0
numpy<=2.0.0,>=1.26.4
