numpy>=1.20
pyyaml>=6

[dev]
pytest
ruff

[torch]
torch>=2.0
torch-geometric>=2.3
