jraph
jraph copied to clipboard
google-deepmind
A Graph Neural Network Library in Jax
I've added a simple function (together with tests) to convert a scipy sparse matrix into a jraph GraphsTuple as per mentioned in issue #36. Let me know if anything need...
Say you want to differentiate edge/node types explicitly, and have differently parametrized functions operate on each type. This is different from types implicitly being encoded in the input embeddings, because...
I couldn't find any function in the jraph API that allows me to convert an arbitrary adjacency matrix to a GraphsTuple object. I assume we are expected to do this...
### Expected behavior The link to [`jraph`'s documentation](https://jraph.readthedocs.io/en/latest/) should take me to a Read the Docs website. ### Experienced behavior Unfortunately, am bouncing between a timeout and an HTML-only formatted...
Jraph is still using `jax.tree_multimap`, which is giving a deprecation warning. This can be problematic for users, for instance for us (Flax), since our CI fails if we hit a...
Hi everyone, I'm trying to re-implement the [GMN model](https://github.com/deepmind/deepmind-research/tree/master/graph_matching_networks) from deepmind in Jax. This model is designed to compute the similarity score between two graphs, and needs to compute the...
Hi! I came across the padding function (`pad_with_graphs`) and have a naive question: why do we use an empty graph there? Also, we specify only the number of edges (and...
Suggest that `_replace` ([as used in the example provided in the README](https://github.com/deepmind/jraph#representing-graphs---the-graphstuple)) be promoted to an external function (e.g. `replace`)
Hi, does this library provide utilities for feature engineering on graphs? For example I may want to calculate different node centrality measures or other metrics that are commonly used in...
