search

google-deepmind / jraph

A Graph Neural Network Library in Jax
https://jraph.readthedocs.io/en/latest/
Apache License 2.0
1.37k stars 89 forks source link

posting flax examples? #51

Open mashaan14 opened 7 months ago

mashaan14 commented 7 months ago

Hi, I've used jraph layers in Flax. Is there an examples section that I can contribute to?

GAT in JAX/Flax: https://github.com/mashaan14/YouTube-channel/blob/main/notebooks/2024_03_18_jraph_GAT.ipynb GCN in JAX/Flax: https://github.com/mashaan14/YouTube-channel/blob/main/notebooks/2024_03_21_jraph_GCN.ipynb Code walkthrough: https://youtu.be/O1zGWMEgW7A