Alternatives and similar repositories for xarray_jax

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

• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆14Updated last year
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆22Updated 2 weeks ago
• arviz-devs / xarray-einstats
  Stats, linear algebra and einops for xarray
  ☆67Updated last week
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆65Updated last week
• dfm / jpu
  JAX + Units
  ☆47Updated 2 months ago
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆41Updated 2 weeks ago
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆37Updated 2 weeks ago
• Nicholaswogan / NumbaQuadpack
  Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.
  ☆29Updated last year
• Team-RADDISH / ParticleDA.jl
  ☆20Updated last month
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆70Updated 9 months ago
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆22Updated last week
• dodgebc / jaxkd
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆41Updated last week
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆20Updated 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 11 months ago
• dfm / rtds-action
  Interface ReadTheDocs and GitHub Actions
  ☆48Updated 4 months ago
• 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
  ☆47Updated last month
• dfm / cv
  My CV
  ☆37Updated this week
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt
  ☆61Updated this week
• CliMA / EnsembleKalmanProcesses.jl
  Implements Optimization and approximate uncertainty quantification algorithms, Ensemble Kalman Inversion, and Ensemble Kalman Processes.
  ☆101Updated this week
• ergodicio / adept
  Automatic-Differentiation-Enabled Plasma Transport in JAX
  ☆31Updated this week
• dask-contrib / dask-histogram
  Histograms with task scheduling.
  ☆24Updated 2 weeks ago
• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated 2 years ago
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆34Updated last month
• tomkimpson / RelativisticDynamics.jl
  General Relativistic Orbital Dynamics in Julia
  ☆20Updated 6 months ago
• python-hydro / how_to_write_a_hydro_code
  A Jupyter/python-based introduction to writing a finite-volume hydro code
  ☆31Updated 5 years ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆97Updated last week
• ASEM000 / finitediffX
  Finite difference tools in JAX
  ☆20Updated last year
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆44Updated last week
• nmoteki / ndimsplinejax
  A multidimensional spline interpolation by Google/JAX
  ☆17Updated last year