numpy>=1.21
torch>=1.9.0
