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:
- Efficiently generate and analyse high dimensional data.☆71Updated last year
- JAX + Units☆52Updated last month
- Stats, linear algebra and einops for xarray☆67Updated 3 weeks ago
- Manipulating ragged arrays in an Array API compliant way.☆45Updated last month
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- Unitful Quantities in JAX☆51Updated this week
- Multi-dimensional data arrays with labeled dimensions☆139Updated this week
- xarray data creation by data classes☆81Updated last year
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated last year
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated 2 years ago
- Taylor Swift-inspired color maps☆84Updated 3 months ago
- Coordinates in JAX☆29Updated 3 weeks ago
- Interface ReadTheDocs and GitHub Actions☆48Updated 11 months ago
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆67Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆112Updated 2 weeks ago
- Histograms with task scheduling.☆25Updated last week
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆51Updated 3 months ago
- Numerical quadrature with JAX☆75Updated last week
- Numerical derivatives for Python☆50Updated last week
- Masked versions of array API compatible arrays☆28Updated last month
- Gaussian Mixture Models in Jax☆23Updated 6 months ago
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆35Updated 9 months ago
- PyMC Extras extracted from the "exoplanet" library☆19Updated last week
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆32Updated last year
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Updated 2 months ago
- einsum via batch matrix multiply☆15Updated 2 years ago
- Numba-accelerated statistical distributions☆67Updated 3 weeks ago
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆32Updated this week
- A library for creating lightweight views of matplotlib axes.☆17Updated last month
- This is the extension library for ASDF to provide support for Astropy.☆19Updated 3 weeks ago