numpy>=1.20
torch>=2.0.0
