U-Cast (Rose-STL-Lab, ICML 2026)

github.com/rose-stl-lab/u-cast
Active41updated 4 months ago
Python
Apache-2.0

Surprisingly simple and efficient frontier probabilistic weather forecaster built on a standard U-Net trained with deterministic MAE pre-training followed by short CRPS fine-tuning via Monte Carlo Dropout, matching or exceeding the probabilistic skill of GenCast and IFS ENS at 1.5° resolution with >10× less training compute than leading CRPS models and >10× lower inference latency than diffusion models; trains in under 12 H200 GPU-days and generates a 15-day ensemble forecast in 3 seconds, with official code and pretrained checkpoints (UCLA, 41+ stars, Apache 2.0)

Sourced from

  • Awesome AI for Science — github.com/rose-stl-lab/u-cast
  • GitHub — github.com/rose-stl-lab/u-cast

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.6K2 months 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.1K4 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.4K2 days 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.2K6 days ago
Python
Apache-2.0