numpy>=1.24.4
torch>=2.1.0
