haiku-geometric (pypi): A collection of graph neural networks implementations in JAX
Keywords: deep-learning|geometric-deep-learning|graph-neural-networks|jax|machine-learning|python