The Encoder and Decoder are single message-passing interaction networks on the grid2mesh and mesh2grid graphs.
Possible Implementation
We can build the interaction network using PyG and call them from a "Denoiser" module, which will handle concatenating input features and graph features (and some noise rescaling as described in the paper).
Implement Encoder and Decoder from GenCast.
Detailed Description
The Encoder and Decoder are single message-passing interaction networks on the grid2mesh and mesh2grid graphs.
Possible Implementation
We can build the interaction network using PyG and call them from a "Denoiser" module, which will handle concatenating input features and graph features (and some noise rescaling as described in the paper).