torch>=2.0
numpy
pyyaml
