torch>=2,<3
numpy>1.22,<2