google-deepmind / xarray_jaxView external linksLinks
☆31Nov 14, 2025Updated 3 months ago
Alternatives and similar repositories for xarray_jax
Users that are interested in xarray_jax are comparing it to the libraries listed below
Sorting:
- Implementation of Computer Vision Models in JAX (equinox)☆18Jan 15, 2026Updated 3 weeks ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆24Nov 13, 2025Updated 3 months ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Oct 28, 2025Updated 3 months ago
- Various tools for creating and manipulating Zarr V3 data with the zarrs Rust crate☆32Feb 1, 2026Updated last week
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆32May 12, 2023Updated 2 years ago
- Python API for the EAGLE cosmological simulation database.☆12Mar 15, 2023Updated 2 years ago
- Hyper Model helps take Machine Learning to production☆10Feb 15, 2021Updated 4 years ago
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated 11 months ago
- ☆49Updated this week
- Implementation of multiplex Leiden for analysis of spatial omics data☆17Feb 2, 2026Updated last week
- Repository for Angus H Wright's AstroStatistics Lecture Materials☆15Apr 6, 2022Updated 3 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Jul 15, 2025Updated 6 months ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 4 months ago
- Paris multilayer transport network☆11Sep 13, 2021Updated 4 years ago
- 📽 Project demonstration generated with the dsproject extension of PyScaffold☆11Jun 6, 2022Updated 3 years ago
- Real-time MIDI fuzzy chord and scale identification☆15Nov 8, 2023Updated 2 years ago
- A Python package to facilitate synthesis and analysis of Earth system data.☆10May 5, 2025Updated 9 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Jan 11, 2025Updated last year
- SO Likelihoods and Theories☆15Feb 3, 2026Updated last week
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- A plumber example and discussion for meetups☆10Jun 6, 2018Updated 7 years ago
- Learn Bayesian Regression on simulated dataset☆10Apr 25, 2018Updated 7 years ago
- numcodecs Rust API for buffer compression☆12Updated this week
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- Collection of Sphinx extensions similar to (but more flexible than) numpydoc☆11Jan 27, 2026Updated 2 weeks ago
- Code showing how to use a model based on the ML model base class.☆10Sep 30, 2022Updated 3 years ago
- interactively bumping SemVer git tags☆10Nov 21, 2020Updated 5 years ago
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 8 months ago
- Pangeo for the European Open Science cloud☆13Feb 6, 2026Updated last week
- files from cl directory of parcftp about 2014☆12Jan 16, 2024Updated 2 years ago
- Earth System Model Collection specification☆13Feb 3, 2023Updated 3 years ago
- Minibatch loading for on-disk anndata files, co-developed by Lamin Labs and scverse☆24Updated this week
- Benchmarking field-level cosmological inference from galaxy surveys.☆12Jul 17, 2025Updated 6 months ago
- MITgcm simulation of an ocean gyre in a rotating tank.☆11Feb 19, 2022Updated 3 years ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated last year
- First Street API Access (Python)☆13Nov 23, 2021Updated 4 years ago
- Build contrasts for models defined with formulaic☆12Updated this week
- port of the VS-Lite model to R.☆14Apr 11, 2019Updated 6 years ago