Skip to content
#

jraph

Here are 6 public repositories matching this topic...

Language: All
Filter by language

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.

  • Updated Aug 14, 2026
  • Python

Add this topic to your repo

To associate your repository with the jraph topic, visit your repo's landing page and select "manage topics."

Learn more