jraph

A Graph Neural Network Library in Jax

1k stars

40 watching

88 forks

Language: Python

last commit: 6 months ago

Linked from 2 awesome lists

deep-learninggraph-neural-networksjaxmachine-learning

Screenshot of google-deepmind/jraph website

https://jraph.readthedocs.io/en/latest/

Backlinks from these awesome lists: