numpy>=1.21.0
torch>=2.0.0
