torch<3.0.0,>=2.0.1
numpy<2.0.0,>=1.26.4
