jraph

Star

Here are 4 public repositories matching this topic...

salfaris / vgae-jax

Star

Variational Graph Autoencoder implemented using Jax & Jraph

machine-learning deep-learning haiku variational-autoencoder jax graph-neural-network optax jraph

Updated Mar 22, 2024
Python

salfaris / PriorVGAE

Star

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

python machine-learning deep-learning networkx gaussian-processes haiku variational-autoencoder jax graph-neural-network numpyro optax jraph

Updated Apr 16, 2023
Python

Zane-Bjornerud / Graphcast-Simplified-GNN-Weather-Forecasting

Star

Simplified, reproducible reimplementation of DeepMind's GraphCast weather model using graph neural networks (JAX · Haiku · Jraph). Runs on a single GPU, with honest negative-results reporting and detailed failure-mode analysis of a heavily scaled-down forecasting model.

machine-learning deep-learning reproducibility message-passing haiku jax graph-neural-networks weather-forecasting gnn autoregressive-models weatherbench jraph graphcast icosahedral-mesh