posting flax examples?

[mashaan14]

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