Hi, is this library compatible with [graph_nets](https://github.com/deepmind/graph_nets) models?
Hi,
is this library compatible with graph_nets models?