Alternatives and similar repositories for xarray_jax:

Users that are interested in xarray_jax are comparing it to the libraries listed below

• arviz-devs / xarray-einstats
  Stats, linear algebra and einops for xarray
  ☆63Updated 3 months ago
• dfm / jpu
  JAX + Units
  ☆46Updated last month
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆36Updated last month
• francois-rozet / dawgz
  Unleash the true power of scheduling
  ☆28Updated 3 weeks ago
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆17Updated this week
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆53Updated this week
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆89Updated last month
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆17Updated 4 months ago
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt
  ☆60Updated this week
• RemoteSensingTools / vSmartMOM.jl
  A full end-to-end modular software suite for radiative transfer and related atmospheric analysis
  ☆44Updated last month
• astropenguin / xarray-dataclasses
  xarray data creation by data classes
  ☆77Updated last month
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆161Updated this week
• dask-contrib / dask-histogram
  Histograms with task scheduling.
  ☆22Updated last week
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆169Updated 2 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆15Updated this week
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆31Updated this week
• dfm / rtds-action
  Interface ReadTheDocs and GitHub Actions
  ☆49Updated 2 weeks ago
• ASEM000 / finitediffX
  Finite difference tools in JAX
  ☆17Updated last year
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆68Updated 5 months ago
• HDembinski / numba-stats
  Numba-accelerated statistical distributions
  ☆55Updated last week
• xarray-contrib / pint-xarray
  Interface for using pint with xarray, providing convenience accessors
  ☆109Updated 2 weeks ago
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆19Updated last month
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆13Updated last year
• adrn / jupyterlab-gist-it
  Upload a Jupyter notebook as a Gist with the click of a button.
  ☆20Updated last year
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆31Updated last week
• xarray-contrib / xarray-schema
  Schema validation for Xarray objects
  ☆41Updated 2 weeks ago
• tacaswell / build_the_world
  Scripts to build much of the Scipy ecosystem from source.
  ☆18Updated 2 months ago
• Carreau / velin
  ☆35Updated last month
• smsharma / jax-conditional-flows
  Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
  ☆15Updated 11 months ago