Alternatives and similar repositories for xarray_jax:

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

• dfm / jpu
  JAX + Units
  ☆47Updated 2 months ago
• arviz-devs / xarray-einstats
  Stats, linear algebra and einops for xarray
  ☆64Updated 4 months ago
• pfackeldey / evermore
  differentiable (binned) likelihoods with JAX
  ☆21Updated this week
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆54Updated this week
• dask-contrib / dask-histogram
  Histograms with task scheduling.
  ☆22Updated last week
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆17Updated last month
• dfm / rtds-action
  Interface ReadTheDocs and GitHub Actions
  ☆48Updated last month
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆20Updated 5 months ago
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆41Updated last month
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt
  ☆60Updated last week
• astropenguin / xarray-dataclasses
  xarray data creation by data classes
  ☆77Updated 3 months ago
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆19Updated last month
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆32Updated last month
• data-apis / array-api-strict
  Strict implementation of the Python array API (previously numpy.array_api)
  ☆20Updated this week
• dfm / cv
  My CV
  ☆37Updated this week
• adonath / gmmx
  Gaussian Mixture Models in Jax
  ☆16Updated 2 months ago
• adrn / jupyterlab-gist-it
  Upload a Jupyter notebook as a Gist with the click of a button.
  ☆20Updated last year
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆169Updated this week
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆90Updated this week
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆69Updated 7 months ago
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆16Updated this week
• scikit-hep / uhi
  Universal Histogram Interface
  ☆18Updated this week
• jpivarski-talks / 2023-07-11-scipy2023-tutorial-thinking-in-arrays
  ☆14Updated last year
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆34Updated this week
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆172Updated 3 months ago
• mdhaber / marray
  Masked versions of array API compatible arrays
  ☆18Updated this week
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• Carreau / velin
  ☆35Updated 2 months ago
• francois-rozet / dawgz
  Unleash the true power of scheduling
  ☆30Updated this week
• xarray-contrib / xarray-schema
  Schema validation for Xarray objects
  ☆42Updated this week