torch>=1.9.0
numpy>=1.21.0
