allen-adastra / xarray_jaxLinks
Simple Xarray + JAX Integration
☆19Updated 7 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:
- Simple differentiable approximate ocean models built with JAX.☆14Updated last year
- Coordinates in JAX☆22Updated 2 weeks ago
- Stats, linear algebra and einops for xarray☆67Updated last week
- Numerical quadrature with JAX☆65Updated last week
- JAX + Units☆47Updated 2 months ago
- Unitful Quantities in JAX☆41Updated 2 weeks ago
- JAX bindings for the NVIDIA cuDecomp library