allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆21Updated last month
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 this week
- Simple differentiable approximate ocean models built with JAX.☆16Updated 2 years ago
- Histograms with task scheduling.☆25Updated last week
- xarray data creation by data classes☆81Updated 9 months ago
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated 2 years ago
- Efficiently generate and analyse high dimensional data.☆71Updated last year
- Interface ReadTheDocs and GitHub Actions☆48Updated 8 months ago
- Manipulating ragged arrays in an Array API compliant way.☆45Updated this week
- JAX + Units☆48Updated last week
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆50Updated last week
- ☆14Updated last year
- Xarray extension and framework for computer model simulations☆75Updated 3 years ago
- Interface for using pint with xarray, providing convenience accessors☆113Updated last week
- A library for creating lightweight views of matplotlib axes.☆17Updated last week
- Advanced algorithms for xarray☆39Updated 7 months ago
- Rotating Shallow Water Model in Python☆20Updated 7 years ago
- ☆35Updated 9 months ago
- Masked versions of array API compatible arrays☆28Updated last week
- Scripts to build much of the Scipy ecosystem from source.☆18Updated 2 months ago
- A xarray extension to show velocity fields as interactive maps in jupyterlab☆12Updated 4 years ago
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆30Updated this week
- A profiler for Numba☆84Updated 8 months ago
- Taylor Swift-inspired color maps☆84Updated last month
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆33Updated 6 months ago
- Numerical quadrature with JAX☆72Updated this week
- Multi-dimensional data arrays with labeled dimensions☆136Updated this week
- Unitful Quantities in JAX☆45Updated this week
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆73Updated last week
- Schema validation for Xarray objects☆42Updated 3 weeks ago
- Fast, flexible, label-aware histograms for numpy and xarray☆95Updated last year