pyyaml>=6.0
torch>=2.2
