neuralgcm / coordaxLinks
Coordinate Axes for JAX
☆59Updated 2 weeks ago
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☆80Updated 2 years ago
- ☆62Updated 2 months ago
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.☆91Updated last week
- Distributed Xarray with Apache Beam☆165Updated last week
- The ✨Magical✨ JAX ML Library.☆18Updated 11 months ago
- Simple Xarray + JAX Integration☆22Updated 4 months ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆34Updated 6 months ago
- Literature and light wrappers for gaussian process models.☆47Updated 4 years ago
- A framework for composing Neural Processes in Python☆89Updated last year
- ☆15Updated last week
- ☆63Updated 4 months ago
- Finite difference tools in JAX☆23Updated last year
- ☆27Updated 4 months ago
- A modular framework for evaluating weather forecasts☆174Updated last week
- A Python package for tackling diverse environmental prediction tasks with NPs.☆120Updated last week
- Find and load data from the Brightband AI-ready mirror of the NOAA NASA Joint Archive (NNJA) of Observations for Earth System Reanalysis☆64Updated 3 weeks ago
- 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 last month
- Stats, linear algebra and einops for xarray☆67Updated 2 weeks ago
- Neural Emulator Architectures in JAX.☆23Updated last year
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆88Updated last week
- Save multichannel data from xarray datasets as videos to save up massive amounts of space☆96Updated 6 months ago
- ☆129Updated 2 weeks ago
- Probabilistic deep learning using JAX☆15Updated 11 months ago
- Benchmarking tools for applying AI/ML to data assimilation☆26Updated 7 months ago
- An open large-scale dataset for training high-resolution physics emulators in hybrid multi-scale climate simulators.☆159Updated 11 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆52Updated 6 months ago
- Gaussian Mixture Models in Jax☆23Updated 6 months ago
- Matrix-free linear algebra in JAX.☆160Updated last month