• License: Apache-2.0
• Home: https://github.com/deepmind/jraph
• Development: https://github.com/deepmind/jraph
• Documentation: https://jraph.readthedocs.io/
• 6876 total downloads
• Last upload: 1 year and 8 months ago

Description

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.