Alternatives and similar repositories for aurora-jax

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

• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated 2 years 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
• alan-turing-institute / mogp-emulator
  Package for fitting Gaussian Process Emulators to multiple output computer simulation results.
  ☆52Updated 2 years ago
• francois-rozet / sda
  Official implementation of Score-based Data Assimilation
  ☆74Updated last year
• googleinterns / invobs-data-assimilation
  ☆38Updated 4 years ago
• jejjohnson / gp_model_zoo
  Literature and light wrappers for gaussian process models.
  ☆47Updated 4 years ago
• stefanosele / GPfY
  Gaussian processes with spherical harmonic features in JAX
  ☆15Updated 3 months ago
• Ceyron / pdequinox
  Neural Emulator Architectures in JAX.
  ☆21Updated last year
• zezhongzhang / EnSF
  Code repository for paper: An ensemble score filter for tracking high-dimensional nonlinear dynamical systems
  ☆20Updated last year
• google-research / swirl-jatmos
  Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.
  ☆34Updated 7 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆53Updated 3 weeks ago
• louity / qgm_pytorch
  PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.
  ☆13Updated 3 years ago
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆17Updated 2 years ago
• mingdeyu / DGP
  Python package 'dgpsi' for deep and linked Gaussian process emulations
  ☆28Updated last week
• duncanwp / ESEm
  Easy emulating of geophysical models including (but not limited to!) Earth System Models.
  ☆59Updated 9 months ago
• dirmeier / ramsey
  Probabilistic deep learning using JAX
  ☆15Updated 10 months ago
• neuralgcm / coordax
  Coordinate Axes for JAX
  ☆36Updated this week
• frazane / scoringrules
  Lightweight library for probabilistic forecast evaluation and optimization.
  ☆67Updated 2 months ago
• modons / DL-weather-dynamics
  ☆21Updated last year
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 4 years ago
• dg227 / NLSA
  Kernel methods for statistical modeling of dynamical systems
  ☆26Updated last month
• ai2cm / fv3net
  explore the FV3 data for parameterization
  ☆17Updated 11 months ago
• wesselb / neuralprocesses
  A framework for composing Neural Processes in Python
  ☆89Updated 11 months ago
• Cambridge-ICCS / practical-ml-with-pytorch
  Training material on writing machine learning code with PyTorch by ICCS
  ☆38Updated 3 months ago
• schmidtjonathan / probabilistic-ssm
  Code accompanying the NeurIPS 2021 Paper: A Probabilistic State Space Model for Joint Inference from Differential Equations and Data (Sch…
  ☆13Updated 3 years ago
• AdrienCorenflos / particle_mala
  Gradient-informed particle MCMC methods
  ☆12Updated last year
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆152Updated last week
• tum-pbs / PhiML
  Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
  ☆93Updated last month
• jswhit / sqgturb
  Python codes for studying predictability and data assimlation with a surface quasi-geostrophic turbulence model
  ☆37Updated this week
• allen-adastra / xarray_jax
  Simple Xarray + JAX Integration
  ☆22Updated 2 months ago