akshatkaush / aurora-jaxLinks
☆25Updated 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☆72Updated last year
- ☆21Updated last year
- Code repository for paper: An ensemble score filter for tracking high-dimensional nonlinear dynamical systems☆18Updated 11 months ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆26Updated 3 months ago
- Easy emulating of geophysical models including (but not limited to!) Earth System Models.☆59Updated 7 months ago
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆33Updated 6 months ago
- ☆38Updated 4 years ago
- Package for fitting Gaussian Process Emulators to multiple output computer simulation results.☆52Updated 2 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11Updated 2 years ago
- Kernel methods for statistical modeling of dynamical systems☆26Updated 2 weeks ago
- Training material on writing machine learning code with PyTorch by ICCS☆37Updated last month
- explore the FV3 data for parameterization☆17Updated 9 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
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆27Updated last week
- How to predict extreme events in climate using rare event algorithms and modern tools of machine learning☆23Updated 7 months ago
- Literature and light wrappers for gaussian process models.☆47Updated 4 years ago
- Simple differentiable approximate ocean models built with JAX.☆16Updated 2 years ago
- Implements Optimization and approximate uncertainty quantification algorithms, Ensemble Kalman Inversion, and Ensemble Kalman Processes.☆109Updated 2 weeks ago
- High Performance Computing for Weather and Climate☆42Updated 3 weeks ago
- OceanBench - SSH edition☆21Updated last year
- Probabilistic deep learning using JAX☆15Updated 8 months ago
- PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.☆13Updated 3 years ago
- Gaussian processes with spherical harmonic features in JAX☆15Updated 2 months ago
- Simple Xarray + JAX Integration☆21Updated last month
- A Highly-Extensible Data Assimilation Testing Suite☆10Updated 6 years ago
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆30Updated last week
- NeuralGCM's differentiable dycore☆303Updated last week
- Bootcamp notebooks☆60Updated 2 months ago
- Lightweight library for probabilistic forecast evaluation and optimization.☆63Updated 3 weeks ago
- A framework for composing Neural Processes in Python☆86Updated 10 months ago