Stormer (Argonne National Lab & U Chicago, NeurIPS 2024)

github.com/tung-nd/stormer
Idle86updated 1 year ago
Python
MIT

Minimal-modification vision transformer for skillful and reliable medium-range weather forecasting, introducing weather-specific patch embedding, randomized dynamics forecasting over varying time intervals, and pressure-weighted loss; competitive at short range and outperforming prior methods beyond 7 days on WeatherBench 2 with orders-of-magnitude less training data and compute, with favorable scaling in model size and training tokens (MIT License)

Sourced from

  • Awesome AI for Science — github.com/tung-nd/stormer
  • GitHub — github.com/tung-nd/stormer

Related resources

Family of operational-quality open weather models from DeepMind and Google Research, including WeatherNext Graph (deterministic GNN medium-range forecasting, published as GraphCast), WeatherNext Gen (diffusion ensemble, published as GenCast), WeatherNext 2 (state-of-the-art global medium-range and cyclone forecasting skillful beyond 15 days, operational at 0.25° resolution), and WeatherNext Cyclones (breakthrough tropical cyclone track forecasting, Nature 2026); official code and open weights (Apache 2.0, 7.6K+ stars)

Active7.7K1 month ago
Python
Apache-2.0

Google DeepMind's diffusion-based ensemble weather forecasting model at 0.25° resolution, outperforming ECMWF ENS on 97.2% of targets up to 15 days ahead, with open-source code and weights (Nature 2024)

Active7.6K1 month ago
Python
Apache-2.0

Python package for segmenting geospatial data with the Segment Anything Model (SAM), enabling zero-shot object segmentation in satellite and aerial imagery for remote sensing and Earth observation (MIT, 4k+ stars)

Active4.1K2 weeks ago
Python
MIT

High-level open-source geospatial AI package for satellite/aerial imagery analysis, model training, inference, interactive visualization, and QGIS integration, bridging PyTorch/Transformers with remote sensing workflows (MIT, 2026)

Active3.3K1 month ago
Python
MIT

Huawei's 3D high-resolution global weather forecast model at 0.25° resolution, first AI method to comprehensively outperform traditional NWP across all variables and lead times, integrated into ECMWF operational forecasts (Nature 2023)

Stale1.4K2 years ago
Python

World's first fully open, accelerated weather AI software stack with Medium Range forecasting and Nowcasting models using generative AI (January 2026)

Active1.1K1 month ago
Python
Apache-2.0