allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆20Updated 8 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 3 weeks ago
- Manipulating ragged arrays in an Array API compliant way.☆44Updated this week
- Efficiently generate and analyse high dimensional data.☆71Updated 10 months ago
- xarray data creation by data classes☆81Updated 6 months ago
- Simple differentiable approximate ocean models built with JAX.☆15Updated last year
- JAX + Units☆47Updated 2 months ago
- Coordinates in JAX☆24Updated 2 weeks ago
- Histograms with task scheduling.☆24Updated this week
- Interface ReadTheDocs and GitHub Actions☆48Updated 5 months ago
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated last year
- Finite difference tools in JAX☆20Updated last year
- differentiable (binned) likelihoods with JAX☆22Updated this week
- Multi-dimensional data arrays with labeled dimensions☆126Updated this week
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆48Updated last month
- ☆14Updated last year
- Gaussian Mixture Models in Jax☆20Updated last week
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt☆61Updated this week
- Interface for using pint with xarray, providing convenience accessors☆111Updated this week
- Masked versions of array API compatible arrays☆23Updated this week
- A library for creating lightweight views of matplotlib axes.☆17Updated 9 months ago
- Unitful Quantities in JAX☆42Updated 2 weeks ago
- Xarray extension and framework for computer model simulations☆74Updated 3 years ago
- Quasigeostrophic model in JAX (port of PyQG)☆19Updated last month
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆20Updated 2 months ago
- My CV☆37Updated this week
- ☆59Updated 2 weeks ago
- Numba-accelerated statistical distributions☆62Updated last week
- Leveraging numba to speed up ODE integration☆73Updated last year
- Some useful extensions for NumPy☆28Updated last year
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆29Updated last year