openclimatefix / graph_weather

PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.07575)
MIT License
187 stars 47 forks source link

Gencast Graphs #110

Closed gbruno16 closed 3 months ago

gbruno16 commented 3 months ago

Implementation of the three graphs in Gencast.

Detailed Description

The mesh is a 6-times refined icosahedral mesh as defined in Lam et al. (2023), with 41,162 nodes and 246,960 edges. Unlike GraphCast, it is not a multi-mesh. Grid2Mesh and Mesh2Grid are directed graphs from the grid to the mesh and in the opposite direction.

Possible Implementation

We can adapt the GraphCast implementation by replacing JAX with NumPy, Torch, and PyG. Furthermore, we may need to implement the 32-hop neighbors mesh.