Comment by EricLeer

See for instance the pytorch geometric [1] package, which is the main implementation in pytorch. They also link to some papers there that might explain you more.

[1] https://pytorch-geometric.readthedocs.io/en/latest/