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
Sorting:
- Rotating Shallow Water Model in Python☆19Updated 7 years ago
- xarray accessor and zarr codec for working with video data☆16Updated last year
- Analysis and visualisation of atmospheric model output powered by iris.☆16Updated this week
- 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
- A ML-oriented generic patcher for xarray data structures.☆17Updated 2 months ago
- explore the FV3 data for parameterization☆17Updated 4 months ago
- A python package for computing zonal mean and related atmospheric diagnostics☆15Updated last year
- Tool for writing large xarray datasets to zarr stores with independent processes☆30Updated last month
- Neural Networks based unified physics parameterization for atmospheric models☆4Updated last year
- Implementations of two-layer models by Held et al. and Geoffroy et al.☆9Updated last year
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆16Updated 3 weeks ago
- Simple Xarray + JAX Integration☆18Updated 6 months ago
- Forecast Sensitivity to Observations and Observation Impact☆18Updated 3 years ago
- Intake catalogue for EUREC4A field campaign datasets☆17Updated 3 months ago
- Interface for using cubed with xarray☆28Updated last week
- ☆14Updated 3 weeks ago
- Python line-by-line radiative transfer model☆11Updated last year
- PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.☆11Updated 3 years ago
- Project Pythia Cookbook template☆26Updated last month
- Toolbox to use NNs in fortran☆12Updated 2 years ago
- Figures with precise control over overall width, plot aspect ratio, between-plot spacing, and colorbar dimensions.☆15Updated 3 years ago
- Continuous wavelet transform with xarray.☆16Updated last year
- MVP for opening xarray datasets from cloud zarr stores in a single thread with asyncio loops☆24Updated last year
- Benchmarking tools for applying AI/ML to data assimilation☆15Updated 2 weeks ago
- A climate change scenario-building analysis framework.☆20Updated this week
- Python wrapper for the FV3-based global climate model☆28Updated 3 years ago
- Project Pythia cookbook for Kerchunk☆16Updated 5 months ago
- Cubed-Sphere data processing with xarray☆18Updated 5 years ago
- Various converters for getting obs data in and out of IODA☆10Updated this week
- Re-write of FV3GFS weather/climate model in Python☆18Updated 2 weeks ago