numpy>=1.24,<3
torch>=2.6,<3
