allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆22Updated 4 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☆67Updated this week
- Simple differentiable approximate ocean models built with JAX.☆16Updated 2 years ago
- JAX + Units☆52Updated last month
- Interface ReadTheDocs and GitHub Actions☆48Updated 11 months ago
- Efficiently generate and analyse high dimensional data.☆71Updated this week
- Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab☆32Updated this week
- Manipulating ragged arrays in an Array API compliant way.☆45Updated last month
- Matplotlib stylesheets to replicate old scientific figure styles☆45Updated 2 years ago
- GCM Physics written in JAX☆79Updated this week
- xarray data creation by data classes☆80Updated last year
- Histograms with task scheduling.☆25Updated last week
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆35Updated 9 months ago
- This repository contains material for the class "Instabilities in Fluids" taught in 2018 at the Australian National University.☆25Updated 2 years ago
- High Performance Computing for Weather and Climate☆42Updated last week
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆67Updated 2 weeks ago
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated last year
- Unitful Quantities in JAX☆52Updated last week
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆92Updated this week
- A Python toolkit for coordinating asynchronous and dynamic ensembles of calculations.☆75Updated this week
- explore the FV3 data for parameterization☆18Updated last year
- My CV☆39Updated this week
- Numerical quadrature with JAX☆75Updated 3 weeks ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆35Updated this week
- Coordinates in JAX☆29Updated last week
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆32Updated last year
- A full end-to-end modular software suite for radiative transfer and related atmospheric analysis☆51Updated last week
- Multi-dimensional data arrays with labeled dimensions☆142Updated this week
- Python wrapper of LSODA (solving ODEs) which can be called from within numba functions.☆103Updated 2 years ago
- Example problems from the Dedalus methods paper.☆18Updated 5 years ago
- A single column model of atmospheric boundary layer turbulence and clouds☆14Updated 5 years ago