Jraph

Introduction to Graph Neural Networks with JAX/jraph

DeepMind's interactive introduction to Graph Neural Networks using JAX and jraph — covers message passing, graph classification, and node prediction with runnable code. One of the clearest practical GNN tutorials available as a Google Colab notebook.