allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆23Updated 2 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☆66Updated 2 weeks ago
- Simple differentiable approximate ocean models built with JAX.☆17Updated 2 years ago
- JAX + Units☆48Updated this week
- Manipulating ragged arrays in an Array API compliant way.☆45Updated this week
- xarray data creation by data classes☆81Updated 10 months ago
- Efficiently generate and analyse high dimensional data.☆71Updated last year
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆50Updated last month
- Multi-dimensional data arrays with labeled dimensions☆136Updated this week
- Numerical quadrature with JAX☆73Updated this week
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆34Updated 7 months ago
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆31Updated last week
- Taylor Swift-inspired color maps☆84Updated last month
- A library for creating lightweight views of matplotlib axes.☆17Updated this week
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated 2 years ago
- Interface ReadTheDocs and GitHub Actions☆48Updated 9 months ago
- Histograms with task scheduling.☆25Updated last week
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated last year
- My CV☆38Updated last week
- Interpolation and function approximation with JAX☆231Updated this week
- Unitful Quantities in JAX☆46Updated last week
- Masked versions of array API compatible arrays☆28Updated this week
- Numba-accelerated statistical distributions☆65Updated 2 weeks ago
- Coordinates in JAX☆27Updated last week
- Interface for using pint with xarray, providing convenience accessors☆113Updated last week
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆64Updated this week
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆31Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11Updated 2 years ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆34Updated this week
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆76Updated this week
- Python wrapper of LSODA (solving ODEs) which can be called from within numba functions.☆102Updated 2 years ago