akshatkaush / aurora-jaxLinks
☆27Updated 4 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:
- ☆38Updated 4 years ago
- Package for fitting Gaussian Process Emulators to multiple output computer simulation results.☆53Updated 2 years ago
- Official implementation of Score-based Data Assimilation☆80Updated 2 years ago
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆35Updated 9 months ago
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- ☆22Updated last year
- Code repository for paper: An ensemble score filter for tracking high-dimensional nonlinear dynamical systems☆20Updated last year
- PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.☆14Updated 3 years ago
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆28Updated last month
- Kernel methods for statistical modeling of dynamical systems☆26Updated 3 months ago
- Easy emulating of geophysical models including (but not limited to!) Earth System Models.☆59Updated 10 months ago
- Python codes for studying predictability and data assimlation with a surface quasi-geostrophic turbulence model☆39Updated this week
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11Updated 2 years ago
- explore the FV3 data for parameterization☆17Updated last year
- Neural Emulator Architectures in JAX.☆23Updated last year
- Literature and light wrappers for gaussian process models.☆47Updated 4 years ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆34Updated 6 months ago
- ☆15Updated last week
- Bootcamp notebooks☆62Updated 5 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
- Gaussian processes with spherical harmonic features in JAX☆15Updated 4 months ago
- A framework for composing Neural Processes in Python☆89Updated last year
- Training material on writing machine learning code with PyTorch by ICCS☆38Updated 4 months ago
- Flood Mapping Intercomparison☆13Updated 2 months ago
- Practicum on Supervised Learning in Function Spaces☆33Updated 3 years ago
- Differentiable neural-network solver for data assimilation of ice shelves written in JAX☆39Updated 8 months ago
- Experiments demonstrating coupled online learning for machine learning parameterizations☆22Updated 2 years ago
- Finite difference tools in JAX☆23Updated last year
- Coordinate Axes for JAX☆59Updated 2 weeks ago
- High Performance Computing for Weather and Climate☆42Updated last week