(building in public) Jeometric: graph neural networks in Jax
github.com