numpy
torch>=2.1
