neuralgcm / coordaxLinks
Coordinate Axes for JAX
☆36Updated this week
Alternatives and similar repositories for coordax
Users that are interested in coordax are comparing it to the libraries listed below
Sorting:
- Official implementation of Score-based Data Assimilation☆74Updated last year
- Simple Xarray + JAX Integration☆22Updated 2 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
- Probabilistic deep learning using JAX☆15Updated 10 months ago
- Literature and light wrappers for gaussian process models.☆47Updated 4 years ago
- Finite difference tools in JAX☆23Updated last year
- Neural Emulator Architectures in JAX.☆21Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11Updated 2 years ago
- A framework for composing Neural Processes in Python☆89Updated 11 months ago
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆22Updated last year
- ☆27Updated 3 months ago
- ☆25Updated last week
- Gaussian processes with spherical harmonic features in JAX☆15Updated 3 months ago
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆28Updated last week
- Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.☆81Updated last week
- A Python package for tackling diverse environmental prediction tasks with NPs.☆119Updated 2 months ago
- Emulate simulations easily☆106Updated last week
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated 3 weeks ago
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆80Updated this week
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆34Updated 7 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 4 months ago
- Easy emulating of geophysical models including (but not limited to!) Earth System Models.☆59Updated 9 months ago
- NeuralGCM's differentiable dycore☆307Updated last week
- ☆129Updated last year
- Stats, linear algebra and einops for xarray☆67Updated this week
- Stainless neural networks in JAX☆34Updated 3 months ago
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Updated 10 months ago
- An open large-scale dataset for training high-resolution physics emulators in hybrid multi-scale climate simulators.☆159Updated 10 months ago
- Package for fitting Gaussian Process Emulators to multiple output computer simulation results.☆52Updated 2 years ago