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
  ☆65Updated 5 months ago
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆17Updated 2 weeks ago
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt
  ☆61Updated last month
• dfm / rtds-action
  Interface ReadTheDocs and GitHub Actions
  ☆48Updated 2 months ago
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆32Updated 2 weeks ago
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆14Updated last year
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆35Updated this week
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆19Updated last month
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆41Updated last month
• astropenguin / xarray-dataclasses
  xarray data creation by data classes
  ☆78Updated 3 months ago
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆17Updated 2 weeks ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆57Updated 2 weeks ago
• ASEM000 / finitediffX
  Finite difference tools in JAX
  ☆18Updated last year
• dfm / emcee-jax
  An experiment: emcee implemented in JAX
  ☆25Updated 2 years ago
• dfm / cv
  My CV
  ☆37Updated this week
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆27Updated 6 months ago
• RemoteSensingTools / vSmartMOM.jl
  A full end-to-end modular software suite for radiative transfer and related atmospheric analysis
  ☆46Updated last week
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆69Updated 7 months ago
• Nicholaswogan / NumbaQuadpack
  Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.
  ☆28Updated last year
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆173Updated this week
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆90Updated 2 weeks ago
• google / xarray-tensorstore
  ☆57Updated 5 months ago
• mdhaber / marray
  Masked versions of array API compatible arrays
  ☆18Updated this week
• astro-informatics / harmonic
  Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection
  ☆64Updated 2 weeks ago
• pmocz / jax-euler-benchmarks
  A simple JAX solver for the 2D Euler equations for benchmarking
  ☆20Updated 6 months ago
• h3jia / bayesfast
  Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.
  ☆29Updated 2 years ago
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆174Updated 2 weeks ago
• Carreau / velin
  ☆35Updated 3 months ago
• jonaselgammal / GPry
  A Package for Bayesian inference of expensive likelihoods with Gaussian Processes
  ☆28Updated last week