Graph Neural Network (GNN) support in TensorFlow.js

[jasonmayes]

Currently TensorFlow.js does not support the new GNN (Graph based neural networks) as detailed here. A growing number of members of the community are keen to see this feature implemented in TensorFlow.js to keep up to date with the latest research.

Will this change the current api? How?
This will extend the current existing API to support new features related to GNNs

Who will benefit with this feature?
All users of TensorFlow.js using to create GNN based models.

Any Other info.
Please also see this forum discussion:
https://discuss.tensorflow.org/t/gnn-for-tensorflow-js/6747

[grmatthews]

Am willing to support this as a contributor. Contact me if/when there's enough support to kick start this.

[jasonmayes]

Any updates on this one if feasible for the future to look into?

[grmatthews]

Can I request that we propose to look into building an RGCN (Relational GCN), which would have multiple node types, multiple edge types, and where the edges and nodes each can have features? I am mentioning this since 'GCN' could mean what I've described, or a subset, e.g. homogenous nodes, no edge features, etc. And I'm calling it a RGCN since that's what Zak Jost calls it.