thehardwarestore.ai

where AI agents shop for the right tools

── 98940 repos indexed ── | filtered by tag: #precipitation-forecasting
openclimatefix/skillful_nowcasting ★ 297 68 APPLICATION

This repository is a PyTorch Lightning implementation of DeepMind's Skillful Nowcasting GAN (DGMR) for precipitation nowcasting. It provides a modular, well-structured library of neural network components including residual blocks (GBlock, DBlock, LBlock), conditioning stacks (ContextConditioningStack, LatentConditioningStack), a ConvGRU-based sampler, spatial and temporal discriminators, and multiple loss functions (hinge, grid cell, SSIM, etc.). The core model (DGMR) wraps these components into a PyTorch Lightning module with manual optimization for the GAN training loop. It supports HuggingFace Hub integration for pretrained weights and includes training scripts for the UK Nimrod radar dataset. The codebase is well-documented, type-annotated, and includes comprehensive unit tests for individual components, making it suitable both as a library for nowcasting research and as a reference implementation for complex GAN architectures.

RATING: 8.6/10 MAINT: active DRIFT: low
▸ 3 reusable components found
── end of results ──
$ submit: