akshatkaush / aurora-jaxLinks
☆26Updated 2 months ago
Alternatives and similar repositories for aurora-jax
Users that are interested in aurora-jax are comparing it to the libraries listed below
Sorting:
- Official implementation of Score-based Data Assimilation☆74Updated last year
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆34Updated 6 months ago
- Package for fitting Gaussian Process Emulators to multiple output computer simulation results.☆52Updated 2 years ago
- Easy emulating of geophysical models including (but not limited to!) Earth System Models.☆59Updated 8 months ago
- ☆21Updated last year
- ☆38Updated 4 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11Updated 2 years ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆30Updated 4 months ago
- Training material on writing machine learning code with PyTorch by ICCS☆38Updated 2 months ago
- Code repository for paper: An ensemble score filter for tracking high-dimensional nonlinear dynamical systems☆20Updated last year
- High Performance Computing for Weather and Climate☆42Updated last week
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆28Updated 3 weeks ago
- Literature and light wrappers for gaussian process models.☆47Updated 4 years ago
- explore the FV3 data for parameterization☆17Updated 10 months ago
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- Kernel methods for statistical modeling of dynamical systems☆26Updated last month
- A framework for composing Neural Processes in Python☆88Updated 11 months ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Updated last year
- Jupyter notebook tutorials on various machine learning topics☆25Updated 9 months ago
- Experiments demonstrating coupled online learning for machine learning parameterizations☆22Updated 2 years ago
- Python codes for studying predictability and data assimlation with a surface quasi-geostrophic turbulence model☆36Updated 8 months ago
- Lightweight library for probabilistic forecast evaluation and optimization.☆65Updated last month
- OceanBench - SSH edition☆22Updated last year
- Finite difference tools in JAX☆22Updated last year
- PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.☆13Updated 3 years ago
- How to predict extreme events in climate using rare event algorithms and modern tools of machine learning☆24Updated 7 months ago
- Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.☆75Updated this week
- Simple Xarray + JAX Integration☆23Updated 2 months ago
- ☆106Updated last year
- Gaussian processes with spherical harmonic features in JAX☆15Updated 2 months ago