numpy>=1.20
torch>=1.9
