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 2 months ago
• dfm / jpu
  JAX + Units
  ☆44Updated this week
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆35Updated this week
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆16Updated this week
• dfm / rtds-action
  Interface ReadTheDocs and GitHub Actions
  ☆49Updated last year
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆84Updated 3 weeks ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆50Updated this week
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆68Updated 4 months ago
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt
  ☆59Updated this week
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆13Updated last year
• astropenguin / xarray-dataclasses
  xarray data creation by data classes
  ☆74Updated 3 weeks ago
• dask-contrib / dask-histogram
  Histograms with task scheduling.
  ☆22Updated this week
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆19Updated 3 months ago
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆32Updated this week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆148Updated this week
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆164Updated last month
• adrn / jupyterlab-gist-it
  Upload a Jupyter notebook as a Gist with the click of a button.
  ☆20Updated last year
• jonaselgammal / GPry
  A Package for Bayesian inference of expensive likelihoods with Gaussian Processes
  ☆22Updated this week
• dfm / cv
  My CV
  ☆36Updated this week
• oceanum-io / xarray-video
  xarray accessor and zarr codec for working with video data
  ☆15Updated last year
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆29Updated this week
• smsharma / jax-conditional-flows
  Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
  ☆14Updated 10 months ago
• RemoteSensingTools / vSmartMOM.jl
  A full end-to-end modular software suite for radiative transfer and related atmospheric analysis
  ☆44Updated 2 weeks ago
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆12Updated this week
• ASEM000 / finitediffX
  Finite difference tools in JAX
  ☆17Updated 11 months ago
• astropy / asdf-astropy
  This is the extension library for ASDF to provide support for Astropy.
  ☆18Updated this week
• google / xarray-tensorstore
  ☆55Updated 2 months ago
• Nicholaswogan / NumbaQuadpack
  Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.
  ☆27Updated 9 months ago
• pfackeldey / evermore
  differentiable (binned) likelihoods with JAX
  ☆19Updated 2 months ago
• tomkimpson / RelativisticDynamics.jl
  General Relativistic Orbital Dynamics in Julia
  ☆20Updated last month