Stars
Recast WOOF: GPU weather model (regional, hex and global)
A highly readable spectral finite element atmospheric dynamical core with automatic differentiation using JAX and/or PyTorch
Continuous background code review database for agents, work faster and smarter with accountability for every line of generated code.
Worktrunk is a CLI for Git worktree management, designed for parallel AI agent workflows
Open-source, cloud-native transactional tensor storage engine
Ai2 Climate Emulator: fast machine learning models for weather and climate prediction
Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.
Minimal JAX implementation of k-nearest neighbors using a k-d tree.
scores: Metrics for the verification, evaluation and optimisation of forecasts, predictions or models.
Universal Notation for Tensor Operations in Python
The versatile ocean simulator, in pure Python, powered by JAX.
A benchmark for the next generation of data-driven global weather models.
MVP for opening xarray datasets from cloud zarr stores in a single thread with asyncio loops
A dead simple Python package for creating custom JAX pytree objects