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 last week
- Simple differentiable approximate ocean models built with JAX.☆16Updated last year
- Efficiently generate and analyse high dimensional data.☆71Updated last year
- JAX + Units☆48Updated last week
- xarray data creation by data classes☆80Updated 8 months ago
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆28Updated 5 months ago
- Histograms with task scheduling.☆24Updated last week
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆62Updated 2 weeks ago
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆30Updated this week
- Finite difference tools in JAX☆22Updated last year
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆49Updated 3 weeks ago
- Unitful Quantities in JAX☆43Updated 2 weeks ago
- High Performance Computing for Weather and Climate☆42Updated 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
- Interface ReadTheDocs and GitHub Actions☆48Updated 7 months ago
- Interface for using pint with xarray, providing convenience accessors☆112Updated last week
- Manipulating ragged arrays in an Array API compliant way.☆44Updated 2 weeks ago
- ☆14Updated last year
- ☆61Updated last week
- A xarray extension to show velocity fields as interactive maps in jupyterlab☆12Updated 4 years ago
- Masked versions of array API compatible arrays☆26Updated 2 weeks ago
- Fast, flexible, label-aware histograms for numpy and xarray☆94Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆11Updated 2 years ago
- Information between bits and bytes.☆36Updated last year
- Numerical quadrature with JAX☆69Updated last week
- explore the FV3 data for parameterization☆17Updated 8 months ago
- A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.☆57Updated last year
- Coordinates in JAX☆25Updated 2 weeks ago
- Trajectory simulations for point particles in the Ocean & Atmosphere☆87Updated 2 weeks ago
- ☆21Updated 3 weeks ago