Useful references: [https://arxiv.org/abs/1806.01261](https://arxiv.org/abs/1806.01261) [https://github.com/deepmind/graph_nets](https://github.com/deepmind/graph_nets)