Popular GitHub Repositories for Jraph

Variational Graph Autoencoder implemented using Jax & Jraph

Master thesis: JAT (Jraph Attention Networks), a deep learning architecture to predict the potential energy and forces of molecules. Adapts Graph Attention Networks (GATv2) within the Message Passing Neural Networks framework to computational chemistry in JAX

Oxford MSc thesis. PriorVAE with graph convolutional networks for learning locally-aware spatial prior distributions