neuralgcm / coordaxLinks
Coordinate Axes for JAX
☆57Updated last week
Alternatives and similar repositories for coordax
Users that are interested in coordax are comparing it to the libraries listed below
Sorting:
- Simple Xarray + JAX Integration☆22Updated 3 months ago
- Official implementation of Score-based Data Assimilation☆77Updated last year
- ☆62Updated 2 months ago
- Finite difference tools in JAX☆23Updated last year
- ☆27Updated 4 months ago
- Stats, linear algebra and einops for xarray☆67Updated 3 weeks ago
- A framework for composing Neural Processes in Python☆89Updated last year
- Probabilistic deep learning using JAX☆15Updated 10 months ago
- A Python package for tackling diverse environmental prediction tasks with NPs.☆119Updated 2 months ago
- Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.☆90Updated last week
- The ✨Magical✨ JAX ML Library.☆18Updated 11 months ago
- Distributed Xarray with Apache Beam☆161Updated last month
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆32Updated 5 months ago
- Neural Emulator Architectures in JAX.☆23Updated last year
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆85Updated this week
- ☆25Updated last week
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆51Updated 5 months ago
- Literature and light wrappers for gaussian process models.☆47Updated 4 years ago
- An open large-scale dataset for training high-resolution physics emulators in hybrid multi-scale climate simulators.☆159Updated 10 months ago
- ☆205Updated 2 months ago
- ☆63Updated 4 months ago
- Emulate simulations easily☆106Updated this week
- Gaussian Mixture Models in Jax☆23Updated 5 months ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Updated last year
- Gaussian processes with spherical harmonic features in JAX☆15Updated 4 months ago
- Find and load data from the Brightband AI-ready mirror of the NOAA NASA Joint Archive (NNJA) of Observations for Earth System Reanalysis☆64Updated last month
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆22Updated last year
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆28Updated 3 weeks ago
- Batch generation from xarray datasets☆190Updated 3 weeks ago