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
  ☆67Updated this week
• dfm / jpu
  JAX + Units
  ☆47Updated last month
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆19Updated this week
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆14Updated last year
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆61Updated this week
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆40Updated this week
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆36Updated last month
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt
  ☆61Updated this week
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆43Updated last month
• astropenguin / xarray-dataclasses
  xarray data creation by data classes
  ☆80Updated 5 months ago
• RemoteSensingTools / vSmartMOM.jl
  A full end-to-end modular software suite for radiative transfer and related atmospheric analysis
  ☆46Updated 2 weeks ago
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆19Updated 3 weeks ago
• astropy / asdf-astropy
  This is the extension library for ASDF to provide support for Astropy.
  ☆18Updated this week
• dask-contrib / dask-histogram
  Histograms with task scheduling.
  ☆24Updated this week
• Nicholaswogan / NumbaQuadpack
  Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.
  ☆29Updated last year
• pnkraemer / tornadox
  Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…
  ☆21Updated 10 months ago
• dfm / emcee-jax
  An experiment: emcee implemented in JAX
  ☆25Updated 2 years ago
• dodgebc / jaxkd
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆37Updated this week
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆95Updated this week
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆22Updated this week
• dfm / rtds-action
  Interface ReadTheDocs and GitHub Actions
  ☆48Updated 3 months ago
• smsharma / jax-conditional-flows
  Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
  ☆20Updated last year
• matplotlib / matplotview
  A library for creating lightweight views of matplotlib axes.
  ☆17Updated 8 months ago
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆31Updated last month
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆70Updated 9 months ago
• pfackeldey / evermore
  differentiable (binned) likelihoods with JAX
  ☆22Updated this week
• Carreau / velin
  ☆35Updated 4 months ago
• dfm / numpyro-ext
  A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work
  ☆24Updated 4 months ago
• google / xarray-tensorstore
  ☆58Updated 6 months ago
• pasteurlabs / tesseract-core
  Autodiff-native, self-documenting software components for Simulation Intelligence. 📦
  ☆51Updated this week