allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆21Updated last week
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 11 months ago
- Stats, linear algebra and einops for xarray☆67Updated 2 weeks ago
- Simple differentiable approximate ocean models built with JAX.☆16Updated last year
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated last year
- Manipulating ragged arrays in an Array API compliant way.☆44Updated last week
- JAX + Units☆48Updated 2 weeks ago
- Histograms with task scheduling.☆24Updated this week
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆49Updated 3 months ago
- xarray data creation by data classes☆80Updated 7 months ago
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated 11 months ago
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆28Updated 4 months ago
- Interface ReadTheDocs and GitHub Actions☆48Updated 6 months ago
- Interface for using pint with xarray, providing convenience accessors☆112Updated this week
- Leveraging numba to speed up ODE integration☆73Updated last year
- Multi-dimensional data arrays with labeled dimensions☆135Updated last week
- Finite difference tools in JAX☆22Updated last year
- A Python toolkit for coordinating asynchronous and dynamic ensembles of calculations.☆70Updated this week
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆70Updated this week
- Information between bits and bytes.☆36Updated last year
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆30Updated last 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
- A library for creating lightweight views of matplotlib axes.☆17Updated last month
- 3D viewer for ESM datasets☆12Updated last month
- Numerical quadrature with JAX☆68Updated this week
- Interpolation and function approximation with JAX☆213Updated last week
- Numba-accelerated statistical distributions☆62Updated 2 weeks ago
- A profiler for Numba☆78Updated 6 months ago
- Unitful Quantities in JAX☆42Updated last month
- Some useful extensions for NumPy☆29Updated last year
- A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.☆57Updated last year