Alternatives and similar repositories for coordax

Users that are interested in coordax are comparing it to the libraries listed below

• allen-adastra / xarray_jax
  Simple Xarray + JAX Integration
  ☆22Updated 4 months ago
• francois-rozet / sda
  Official implementation of Score-based Data Assimilation
  ☆84Updated 2 years ago
• dirmeier / ramsey
  Probabilistic deep learning using JAX
  ☆15Updated last year
• google / xarray-tensorstore
  ☆64Updated 3 months ago
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆22Updated last year
• wesselb / neuralprocesses
  A framework for composing Neural Processes in Python
  ☆89Updated last year
• neuralgcm / dinosaur
  NeuralGCM's differentiable dycore
  ☆317Updated last week
• akshatkaush / aurora-jax
  ☆27Updated 5 months ago
• Ceyron / pdequinox
  Neural Emulator Architectures in JAX.
  ☆23Updated last year
• pasteurlabs / tesseract-core
  Universal, autodiff-native software components for Simulation Intelligence. 📦
  ☆92Updated this week
• ASEM000 / finitediffX
  Finite difference tools in JAX
  ☆23Updated 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
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆159Updated 2 months ago
• jejjohnson / gp_model_zoo
  Literature and light wrappers for gaussian process models.
  ☆47Updated 4 years ago
• schmidtjonathan / Climate2Weather
  This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…
  ☆36Updated 6 months ago
• adonath / gmmx
  Gaussian Mixture Models in Jax
  ☆24Updated 6 months ago
• dodgebc / jaxkd
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆53Updated 6 months ago
• danielward27 / flowjax
  ☆216Updated last week
• alan-turing-institute / deepsensor
  A Python package for tackling diverse environmental prediction tasks with NPs.
  ☆121Updated last month
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated 2 months ago
• google / xarray-beam
  Distributed Xarray with Apache Beam
  ☆165Updated 3 weeks ago
• leap-stc / ClimSim
  An open large-scale dataset for training high-resolution physics emulators in hybrid multi-scale climate simulators.
  ☆159Updated last year
• brightbandtech / ExtremeWeatherBench
  Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.
  ☆97Updated last week
• ssm-jax / ssm-book
  Interactive textbook on state space models
  ☆12Updated 3 years ago
• arviz-devs / xarray-einstats
  Stats, linear algebra and einops for xarray
  ☆67Updated this week
• ASEM000 / serket
  The ✨Magical✨ JAX ML Library.
  ☆18Updated last year
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆53Updated 2 months ago
• v0lta / Jax-Wavelet-Toolbox
  Differentiable and gpu enabled fast wavelet transforms in JAX.
  ☆48Updated last year
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆16Updated 2 years ago
• alan-turing-institute / mogp-emulator
  Package for fitting Gaussian Process Emulators to multiple output computer simulation results.
  ☆53Updated 2 years ago