numpy
torch>=2.2
