Raymondlol/Faster-WeatherNext

Google DeepMind's WeatherNext 2 at full 0.25° on consumer GPUs. Same model, same results: 6.4 GiB instead of 34 GiB, 7× faster.

语言：Python · GitHub 星标：9 · Fork：2

话题：ensemble-forecasting · flash-attention · gpu · graph-neural-networks · jax · machine-learning · numerical-weather-prediction · pallas

在 GitHub 上查看原仓库