keisler-2022

这是一个基于图神经网络（GNN）的全球天气预报模型实现，复现了Keisler 2022论文提出的三阶段架构：Encoder将ERA5经纬度网格映射至H3六边形图结构，Processor在图上执行9轮消息传递，Decoder映射回原始网格生成6小时预报增量，并通过自回归滚动实现多日预测。项目采用JAX实现，支持CPU/GPU加速，可接入ERA5再分析数据或ECMWF IFS实时分析数据，适用于气象研究、数值预报方法验证及轻量级业务化试验场景。