julien-rsbrg/haiku-geometric-scale-up

A collection of graph neural networks implementations in JAX with tools to handle graphs with a very large number of nodes.

Python