Working with Graph Neural Networks (Varying State Space)

tagging @gjoliver here