jraph (pypi): Jraph: A library for Graph Neural Networks in Jax
Keywords: jax|graph|neural|networks|python|machine|learning|deep-learning|graph-neural-networks|machine-learning