Alternatives and similar repositories for aurora-jax

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

• alan-turing-institute / mogp-emulator
  Package for fitting Gaussian Process Emulators to multiple output computer simulation results.
  ☆51Updated last year
• francois-rozet / sda
  Official implementation of Score-based Data Assimilation
  ☆72Updated last year
• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated 2 years ago
• jejjohnson / gp_model_zoo
  Literature and light wrappers for gaussian process models.
  ☆48Updated 4 years ago
• googleinterns / invobs-data-assimilation
  ☆39Updated 4 years ago
• mingdeyu / DGP
  Python package 'dgpsi' for deep and linked Gaussian process emulations
  ☆27Updated 2 weeks ago
• modons / DL-weather-dynamics
  ☆21Updated last year
• 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
• allen-adastra / xarray_jax
  Simple Xarray + JAX Integration
  ☆21Updated 3 weeks ago
• google-research / swirl-jatmos
  Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.
  ☆28Updated 4 months ago
• duncanwp / ESEm
  Easy emulating of geophysical models including (but not limited to!) Earth System Models.
  ☆59Updated 6 months ago
• dirmeier / ramsey
  Probabilistic deep learning using JAX
  ☆15Updated 7 months ago
• schmidtjonathan / Climate2Weather
  This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…
  ☆24Updated last month
• frazane / scoringrules
  Lightweight library for probabilistic forecast evaluation and optimization.
  ☆63Updated 2 weeks ago
• Ceyron / pdequinox
  Neural Emulator Architectures in JAX.
  ☆19Updated 10 months ago
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆16Updated last year
• wesselb / neuralprocesses
  A framework for composing Neural Processes in Python
  ☆85Updated 9 months ago
• stefanosele / GPfY
  Gaussian processes with spherical harmonic features in JAX
  ☆15Updated 3 weeks ago
• Cambridge-ICCS / practical-ml-with-pytorch
  Training material on writing machine learning code with PyTorch by ICCS
  ☆34Updated last week
• zezhongzhang / EnSF
  Code repository for paper: An ensemble score filter for tracking high-dimensional nonlinear dynamical systems
  ☆16Updated 10 months ago
• louity / qgm_pytorch
  PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.
  ☆13Updated 3 years ago
• dg227 / NLSA
  Kernel methods for statistical modeling of dynamical systems
  ☆26Updated 2 weeks ago
• georgemilosh / Climate-Learning
  How to predict extreme events in climate using rare event algorithms and modern tools of machine learning
  ☆22Updated 5 months ago
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 4 years ago
• tum-pbs / PhiML
  Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
  ☆92Updated this week
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆22Updated last year
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆53Updated last week
• ssm-jax / ssm-book
  Interactive textbook on state space models
  ☆12Updated 3 years ago
• ai2cm / fv3net
  explore the FV3 data for parameterization
  ☆17Updated 8 months ago
• cemac / LIFD_ENV_ML_NOTEBOOKS
  Jupyter notebook tutorials on various machine learning topics
  ☆25Updated 7 months ago