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
  ☆52Updated 3 weeks ago
• arviz-devs / xarray-einstats
  Stats, linear algebra and einops for xarray
  ☆67Updated last week
• jejjohnson / jaxsw
  Simple differentiable approximate ocean models built with JAX.
  ☆17Updated 2 years ago
• astropenguin / xarray-dataclasses
  xarray data creation by data classes
  ☆81Updated last year
• jcmgray / xyzpy
  Efficiently generate and analyse high dimensional data.
  ☆71Updated last year
• dfm / rtds-action
  Interface ReadTheDocs and GitHub Actions
  ☆48Updated 10 months ago
• scikit-hep / ragged
  Manipulating ragged arrays in an Array API compliant way.
  ☆45Updated 3 weeks ago
• RemoteSensingTools / vSmartMOM.jl
  A full end-to-end modular software suite for radiative transfer and related atmospheric analysis
  ☆51Updated 2 months ago
• JBorrow / swiftascmaps
  Taylor Swift-inspired color maps
  ☆84Updated 3 months ago
• scikit-hep / numba-stats
  Numba-accelerated statistical distributions
  ☆65Updated this week
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆50Updated last week
• dask-contrib / dask-histogram
  Histograms with task scheduling.
  ☆25Updated 2 weeks ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆74Updated this week
• eigenfoo / littlemcmc
  A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.
  ☆59Updated last year
• exoplanet-dev / pymc-ext
  PyMC Extras extracted from the "exoplanet" library
  ☆19Updated last month
• ckinzthompson / oldplotlib
  Matplotlib stylesheets to replicate old scientific figure styles
  ☆45Updated 2 years ago
• 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 last year
• open-atmos / PyMPDATA
  Numba-accelerated Pythonic implementation of MPDATA with examples in Python, Julia, Rust and Matlab
  ☆31Updated this week
• xarray-contrib / pint-xarray
  Interface for using pint with xarray, providing convenience accessors
  ☆115Updated this week
• JuliaClimate / Drifters.jl
  Trajectory simulations for point particles in the Ocean & Atmosphere
  ☆93Updated last month
• danielward27 / paramax
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆48Updated 8 months ago
• pymc-devs / sunode
  Solve ODEs fast, with support for PyMC
  ☆117Updated last year
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆29Updated last week
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆109Updated this week
• nmoteki / ndimsplinejax
  A multidimensional spline interpolation by Google/JAX
  ☆19Updated last year
• HajimeKawahara / exojax
  🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs
  ☆66Updated 3 weeks ago
• pythonspeed / profila
  A profiler for Numba
  ☆85Updated 10 months ago
• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated 2 years ago
• casus / minterpy
  Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …
  ☆39Updated last year
• PyRsw / PyRsw
  Rotating Shallow Water Model in Python
  ☆20Updated 8 years ago