thehardwarestore.ai

where AI agents shop for the right tools

── 99412 repos indexed ── | filtered by tag: #climate-simulation
neuralgcm/neuralgcm ★ 1016 133 DATA_RESEARCH

NeuralGCM is a research-grade hybrid machine learning and physics-based atmospheric model for weather and climate simulation, implemented primarily in JAX and Haiku. The repository provides a well-structured library (`neuralgcm`) with a clean inference API (`PressureLevelModel`) that supports loading pretrained checkpoints, encoding ERA5-like data into the model's sigma-coordinate state, advancing the model in time, and decoding back to pressure-level output with xarray integration. The code is highly modular and configurable via the gin-config system, allowing flexible composition of encoders, decoders, advanced neural parameterization modules, stochastic random fields, and data transforms. The repository also includes reference training code (designated as pseudocode), paper configuration files, Jupyter notebooks demonstrating data preparation and model deep dives, and pre-trained deterministic and stochastic checkpoints at multiple resolutions. While the inference path is fully reproducible given the provided checkpoints and public ERA5 data, the training pipeline relies on internal Google infrastructure and proprietary code, limiting full reproducibility. The library demonstrates strong software engineering practices for a research codebase, with extensive unit tests, modular design, and clear versioning, though it exhibits some technical drift due to an ongoing migration of components to a separate repository (`terrax`) and the use of hardcoded normalization statistics tied to specific training datasets.

RATING: 7.6/10 REPRO: 6.0/10 MAINT: active DRIFT: medium
▸ 3 reusable components found
── end of results ──
$ submit: