allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆21Updated 3 weeks 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 2 weeks ago
- Simple differentiable approximate ocean models built with JAX.☆16Updated last year
- Efficiently generate and analyse high dimensional data.☆71Updated last year
- xarray data creation by data classes☆80Updated 9 months ago
- JAX + Units☆48Updated last month
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆30Updated this week
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆30Updated 5 months ago
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated last year
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆49Updated this week
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆70Updated last week
- Taylor Swift-inspired color maps☆84Updated last week
- Manipulating ragged arrays in an Array API compliant way.☆45Updated last month
- Interface ReadTheDocs and GitHub Actions☆48Updated 7 months ago
- Histograms with task scheduling.☆25Updated this week
- This repository contains material for the class "Instabilities in Fluids" taught in 2018 at the Australian National University.☆25Updated 2 years ago
- Interface for using pint with xarray, providing convenience accessors☆113Updated this week
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆33Updated last week
- Rotating Shallow Water Model in Python☆20Updated 7 years ago
- PyMC Extras extracted from the "exoplanet" library☆19Updated this week
- Python wrapper of LSODA (solving ODEs) which can be called from within numba functions.☆100Updated 2 years ago
- Numba-accelerated statistical distributions☆65Updated last week
- PYthon RAndom SAmpling for MEshes☆16Updated this week
- Xarray extension and framework for computer model simulations☆75Updated 3 years ago
- Multi-dimensional data arrays with labeled dimensions☆136Updated this week
- explore the FV3 data for parameterization☆17Updated 9 months ago
- A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.☆57Updated last year
- Finite difference tools in JAX☆22Updated last year
- 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
- Some useful extensions for NumPy☆29Updated last year
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆30Updated last year