Show HN: Haiku Geometric – Graph Neural Networks for JAX

github.com/alexOarga

2 points

oarga

3 years ago

I recently made this graph neural networks library. I haven't found anything similar for JAX. There is Jraph, which I like, but I wanted something more straightforward and with more variety of layers/parameters.