torch>=2.0.0
pyyaml>=6.0

[dev]
pytest>=7.0.0
build
twine
black
