Alternatives and similar repositories for aurora-jax

Users that are interested in aurora-jax are comparing it to the libraries listed below

• francois-rozet / sda
  Official implementation of Score-based Data Assimilation
  ☆72Updated last year
• modons / DL-weather-dynamics
  ☆21Updated last year
• zezhongzhang / EnSF
  Code repository for paper: An ensemble score filter for tracking high-dimensional nonlinear dynamical systems
  ☆18Updated 11 months ago
• schmidtjonathan / Climate2Weather
  This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…
  ☆26Updated 3 months ago
• duncanwp / ESEm
  Easy emulating of geophysical models including (but not limited to!) Earth System Models.
  ☆59Updated 7 months ago
• google-research / swirl-jatmos
  Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.
  ☆33Updated 6 months ago
• googleinterns / invobs-data-assimilation
  ☆38Updated 4 years ago
• alan-turing-institute / mogp-emulator
  Package for fitting Gaussian Process Emulators to multiple output computer simulation results.
  ☆52Updated 2 years ago
• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated 2 years ago
• dg227 / NLSA
  Kernel methods for statistical modeling of dynamical systems
  ☆26Updated 2 weeks ago
• Cambridge-ICCS / practical-ml-with-pytorch
  Training material on writing machine learning code with PyTorch by ICCS
  ☆37Updated last month
• ai2cm / fv3net
  explore the FV3 data for parameterization
  ☆17Updated 9 months ago
• pnkraemer / tornadox
  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
• mingdeyu / DGP
  Python package 'dgpsi' for deep and linked Gaussian process emulations
  ☆27Updated last week
• georgemilosh / Climate-Learning
  How to predict extreme events in climate using rare event algorithms and modern tools of machine learning
  ☆23Updated 7 months ago
• jejjohnson / gp_model_zoo
  Literature and light wrappers for gaussian process models.
  ☆47Updated 4 years ago
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆16Updated 2 years ago
• CliMA / EnsembleKalmanProcesses.jl
  Implements Optimization and approximate uncertainty quantification algorithms, Ensemble Kalman Inversion, and Ensemble Kalman Processes.
  ☆109Updated 2 weeks ago
• ofuhrer / HPC4WC
  High Performance Computing for Weather and Climate
  ☆42Updated 3 weeks ago
• jejjohnson / oceanbench
  OceanBench - SSH edition
  ☆21Updated last year
• dirmeier / ramsey
  Probabilistic deep learning using JAX
  ☆15Updated 8 months ago
• louity / qgm_pytorch
  PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.
  ☆13Updated 3 years ago
• stefanosele / GPfY
  Gaussian processes with spherical harmonic features in JAX
  ☆15Updated 2 months ago
• allen-adastra / xarray_jax
  Simple Xarray + JAX Integration
  ☆21Updated last month
• a-attia / DATeS
  A Highly-Extensible Data Assimilation Testing Suite
  ☆10Updated 6 years ago
• open-atmos / PyMPDATA
  Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab
  ☆30Updated last week
• neuralgcm / dinosaur
  NeuralGCM's differentiable dycore
  ☆303Updated last week
• neuraloperator / bootcamp
  Bootcamp notebooks
  ☆60Updated 2 months ago
• frazane / scoringrules
  Lightweight library for probabilistic forecast evaluation and optimization.
  ☆63Updated 3 weeks ago
• wesselb / neuralprocesses
  A framework for composing Neural Processes in Python
  ☆86Updated 10 months ago