jejjohnson / jaxsw
Simple differentiable approximate ocean models built with JAX.
☆14Updated last year
Alternatives and similar repositories for jaxsw:
Users that are interested in jaxsw are comparing it to the libraries listed below
- Rotating Shallow Water Model in Python☆19Updated 7 years ago
- Python package based on PyTorch and Dask for efficient and parallelized data assimilation of weather model data and observations on GPUs.☆12Updated 2 months ago
- Figures with precise control over overall width, plot aspect ratio, between-plot spacing, and colorbar dimensions.☆15Updated 3 years ago
- Implementations of two-layer models by Held et al. and Geoffroy et al.☆9Updated last year
- Toolbox to use NNs in fortran☆12Updated 2 years ago
- Analysis and visualisation of atmospheric model output powered by iris.☆16Updated last week
- Neural Networks based unified physics parameterization for atmospheric models☆4Updated last year
- A python package for computing zonal mean and related atmospheric diagnostics☆15Updated last year
- Continuous wavelet transform with xarray.☆15Updated last year
- A climate change scenario-building analysis framework.☆20Updated this week
- A ML-oriented generic patcher for xarray data structures.☆17Updated 2 months ago
- Python remapping tools for climate and earth system models.☆19Updated last week
- xarray accessor and zarr codec for working with video data☆16Updated last year
- PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.☆11Updated 2 years ago
- A lightweight machine learning framework for Xarray☆22Updated last year
- 2022 EarthCube Annual Meeting☆11Updated 2 years ago
- Project Pythia cookbook for Kerchunk☆16Updated 4 months ago
- Statistical and visual tools for gathering metrics and comparing Earth System Model data files. A common use case is comparing data that…☆19Updated this week
- explore the FV3 data for parameterization☆17Updated 3 months ago
- Various converters for getting obs data in and out of IODA☆10Updated this week
- Intake catalogue for EUREC4A field campaign datasets☆17Updated 2 months ago
- A Highly-Extensible Data Assimilation Testing Suite☆10Updated 6 years ago
- Pythie is a Python package to postprocess weather forecasts. Presently, it contains the modules: MBM: a module to make member-by-member e…☆11Updated 2 years ago
- Tool for writing large xarray datasets to zarr stores with independent processes☆28Updated 2 weeks ago
- Cubed-Sphere data processing with xarray☆18Updated 5 years ago
- code for simple models of the atmosphere and ocean☆30Updated 3 years ago
- A 2-layer quasi-geostrophic atmospheric model in Python. Can be coupled to a simple land or shallow-water ocean component.☆39Updated 3 weeks ago
- Regrid global ocean data in 3d.☆11Updated 2 weeks ago
- raster tools for xarray☆16Updated this week
- Python line-by-line radiative transfer model☆11Updated last year