numpy>=1.25
torch>=2.2
