jraph (conda): Jraph (pronounced "giraffe") is a lightweight library for working with graph neural networks in jax. It provides a data structure for graphs, a set of utilities for working with graphs, and a 'zoo' of forkable graph neural network models.
Keywords: deep-learning|graph-neural-networks|jax|machine-learning