torch>=2.1.0
PyYAML>=6.0
numpy>=1.23
