allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆22Updated 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:
- Stats, linear algebra and einops for xarray☆67Updated last week
- Efficiently generate and analyse high dimensional data.☆71Updated last year
- JAX + Units☆52Updated this week
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- Interface ReadTheDocs and GitHub Actions☆48Updated 10 months ago
- xarray data creation by data classes☆81Updated 11 months ago
- Unitful Quantities in JAX☆48Updated this week
- Manipulating ragged arrays in an Array API compliant way.☆45Updated last week
- Histograms with task scheduling.☆25Updated this week
- Gaussian Mixture Models in Jax☆23Updated 5 months ago
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated 2 years ago
- Coordinates in JAX☆28Updated 2 weeks ago
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆51Updated 2 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
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆107Updated this week
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆31Updated last week
- Numerical quadrature with JAX☆73Updated this week
- Numba-accelerated statistical distributions☆65Updated 2 weeks ago
- Taylor Swift-inspired color maps☆84Updated 2 months ago
- My CV☆38Updated this week
- Multi-dimensional data arrays with labeled dimensions☆136Updated this week
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Updated last year
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆66Updated this week
- Trajectory simulations for point particles in the Ocean & Atmosphere☆92Updated last week
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆31Updated last year
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Updated last month
- A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.☆59Updated last year
- ☆35Updated 10 months ago
- Interface for using pint with xarray, providing convenience accessors☆114Updated this week
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆82Updated this week