allen-adastra / xarray_jax
Simple Xarray + JAX Integration
☆17Updated 5 months ago
Alternatives and similar repositories for xarray_jax:
Users that are interested in xarray_jax are comparing it to the libraries listed below
- JAX + Units☆47Updated 2 months ago
- Stats, linear algebra and einops for xarray☆64Updated 4 months ago
- differentiable (binned) likelihoods with JAX☆21Updated this week
- Numerical quadrature with JAX☆54Updated this week
- Histograms with task scheduling.☆22Updated last week
- Coordinates in JAX☆17Updated last month
- Interface ReadTheDocs and GitHub Actions☆48Updated last month
- A small library of paramaterizations and parameter constraints for PyTrees.☆20Updated 5 months ago
- Manipulating ragged arrays in an Array API compliant way.☆41Updated last month
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt☆60Updated last week
- xarray data creation by data classes☆77Updated 3 months ago
- PyMC Extras extracted from the "exoplanet" library☆19Updated last month
- JAX bindings for the NVIDIA cuDecomp library☆32Updated last month
- Strict implementation of the Python array API (previously numpy.array_api)☆20Updated this week
- My CV☆37Updated this week
- Gaussian Mixture Models in Jax☆16Updated 2 months ago
- Upload a Jupyter notebook as a Gist with the click of a button.☆20Updated last year
- Interpolation and function approximation with JAX☆169Updated this week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆90Updated this week
- Efficiently generate and analyse high dimensional data.☆69Updated 7 months ago
- Orthogonal polynomials with JAX☆16Updated this week
- Universal Histogram Interface☆18Updated this week
- ☆14Updated last year
- Unitful Quantities in JAX☆34Updated this week
- Probabilistic Programming and Nested sampling in JAX☆172Updated 3 months ago
- Masked versions of array API compatible arrays☆18Updated this week
- Stencil computations in JAX☆70Updated last year
- ☆35Updated 2 months ago
- Unleash the true power of scheduling☆30Updated this week
- Schema validation for Xarray objects☆42Updated this week