numpy<3,>=1.26
torch<3,>=2.1
