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
  ☆66Updated 5 months ago
• dfm / jpu
  JAX + Units
  ☆47Updated 3 weeks ago
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆14Updated last year
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆41Updated 2 weeks ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆58Updated last week
• dask-contrib / dask-histogram
  Histograms with task scheduling.
  ☆24Updated this week
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆36Updated last week
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆17Updated this week
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆29Updated last week
• astropenguin / xarray-dataclasses
  xarray data creation by data classes
  ☆78Updated 4 months ago
• RemoteSensingTools / vSmartMOM.jl
  A full end-to-end modular software suite for radiative transfer and related atmospheric analysis
  ☆46Updated last week
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt
  ☆61Updated this week
• dfm / cv
  My CV
  ☆37Updated this week
• casus / minterpy
  Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …
  ☆39Updated 7 months ago
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆19Updated 2 months ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆93Updated this week
• tacaswell / build_the_world
  Scripts to build much of the Scipy ecosystem from source.
  ☆18Updated this week
• ergodicio / adept
  Automatic-Differentiation-Enabled Plasma Transport in JAX
  ☆31Updated this week
• google / xarray-tensorstore
  ☆57Updated 6 months ago
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆70Updated 8 months ago
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆38Updated this week
• pfackeldey / evermore
  differentiable (binned) likelihoods with JAX
  ☆22Updated this week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆182Updated last week
• xarray-contrib / xarray-schema
  Schema validation for Xarray objects
  ☆42Updated last month
• 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 9 months ago
• Carreau / velin
  ☆35Updated 3 months ago
• mdhaber / marray
  Masked versions of array API compatible arrays
  ☆21Updated 3 weeks ago
• pmocz / jax-euler-benchmarks
  A simple JAX solver for the 2D Euler equations for benchmarking
  ☆20Updated 7 months ago
• xarray-contrib / pint-xarray
  Interface for using pint with xarray, providing convenience accessors
  ☆109Updated this week
• jpivarski-talks / 2023-07-11-scipy2023-tutorial-thinking-in-arrays
  ☆14Updated last year