numpy<2,>=1.19.5
torch<2,>=1
