numpy<2

[torch]
torch==2.4.1
