allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆22Updated 4 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:
- Stats, linear algebra and einops for xarray☆67Updated this week
- Simple differentiable approximate ocean models built with JAX.☆16Updated 2 years ago
- JAX + Units☆52Updated last month
- Efficiently generate and analyse high dimensional data.☆71Updated this week
- Manipulating ragged arrays in an Array API compliant way.☆45Updated last month
- xarray data creation by data classes☆80Updated last year
- Unitful Quantities in JAX☆52Updated last week
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated 2 years ago
- Interface ReadTheDocs and GitHub Actions☆48Updated 11 months ago
- Coordinates in JAX☆29Updated last week
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆67Updated 2 weeks ago
- PyMC Extras extracted from the "exoplanet" library☆19Updated 3 weeks ago
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆32Updated last year
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆51Updated last week
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆35Updated 9 months ago
- A library for creating lightweight views of matplotlib axes.☆17Updated last month
- Interface for using pint with xarray, providing convenience accessors☆115Updated last week
- Numba-accelerated statistical distributions☆67Updated last month
- Gaussian Mixture Models in Jax☆24Updated 6 months ago
- My CV☆39Updated last week
- Histograms with task scheduling.☆25Updated last week
- Numerical quadrature with JAX☆75Updated 3 weeks ago
- Making Fabio Crameri's perceptually uniform colourmaps for geosciences available on PyPI and conda-forge☆208Updated last week
- A profiler for Numba☆87Updated 11 months ago
- ☆64Updated 3 months ago
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆32Updated last week
- Taylor Swift-inspired color maps☆85Updated 4 months ago
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated last year
- A Package for Bayesian inference of expensive likelihoods with Gaussian Processes☆28Updated 2 months ago
- A multidimensional spline interpolation by Google/JAX☆19Updated last year