Is this project still live? about graph_nets HOT 2 OPEN

Thanks for your message @caiodadauto, the library itself is not deprecated but our team shifted to JAX, and as @8bitmp3 mentioned we have also developed a JAX version of the library. That said if you have any contributions for the graph_nets library that are general purpose and you would like to provide, and are happy to clean up the code and add tests, we would be happy to review it and merge it (although before doing the work, perhaps create an issue/feature request for each piece of functionality so we can discuss it!).

We have also advised on the design of TF-GNN which is a new tensorflow library for GNNs and provides similar functionality to the graph_nets library, but also offers native support for graphs with explicitly typed nodes and edges, and is designed around Keras, which is one of the most popular model frameworks for TensorFlow. If you are interested on exploring this, a good starting point would be the shortest path demo which we migrated to TF-GNN.

from graph_nets.

Can't speak for the team but have you seen https://github.com/deepmind/jraph? (copying @jg8610).

from graph_nets.