Alternatives and similar repositories for eqxvision

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

• patrick-kidger / quax

  View on GitHub
  Multiple dispatch over abstract array types in JAX.
  ☆136Dec 15, 2025Updated 2 months ago
• marin-community / haliax

  View on GitHub
  Named Tensors for Legible Deep Learning in JAX
  ☆218Nov 8, 2025Updated 3 months ago
• patrick-kidger / equinox

  View on GitHub
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,771Updated this week
• patrick-kidger / optimistix

  View on GitHub
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆541Updated this week
• ASEM000 / kernex

  View on GitHub
  Stencil computations in JAX
  ☆71Nov 15, 2025Updated 3 months ago
• ASEM000 / pytreeclass

  View on GitHub
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Jan 11, 2025Updated last year
• xl0 / lovely-jax

  View on GitHub
  JAX Arrays for human consumption
  ☆111Oct 23, 2025Updated 3 months ago
• patrick-kidger / lineax

  View on GitHub
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆510Updated this week
• brentyi / jax_dataclasses

  View on GitHub
  Pytrees + dataclasses ❤️
  ☆74Dec 19, 2025Updated last month
• young-geng / scalax

  View on GitHub
  A simple library for scaling up JAX programs
  ☆145Nov 4, 2025Updated 3 months ago
• patrick-kidger / jaxtyping

  View on GitHub
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,730Jan 30, 2026Updated 2 weeks ago
• google / CommonLoopUtils

  View on GitHub
  CLU lets you write beautiful training loops in JAX.
  ☆366Jan 21, 2026Updated 3 weeks ago
• danielward27 / paramax

  View on GitHub
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆50Jan 23, 2026Updated 3 weeks ago
• patrick-kidger / sympy2jax

  View on GitHub
  Turn SymPy expressions into trainable JAX expressions.
  ☆360Feb 4, 2026Updated last week
• patrick-kidger / wadler_lindig

  View on GitHub
  A Wadler–Lindig pretty printer for Python
  ☆44Jan 16, 2026Updated 3 weeks ago
• samuela / torch2jax

  View on GitHub
  Run PyTorch in JAX. 🤝
  ☆312Oct 13, 2025Updated 4 months ago
• pnkraemer / probfindiff

  View on GitHub
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11May 8, 2023Updated 2 years ago
• patrick-kidger / diffrax

  View on GitHub
  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
  ☆1,898Feb 8, 2026Updated last week
• marin-community / levanter

  View on GitHub
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆693Jan 26, 2026Updated 2 weeks ago
• google / tree-math

  View on GitHub
  Mathematical operations for JAX pytrees
  ☆208Dec 5, 2024Updated last year
• DarshanDeshpande / jax-models

  View on GitHub
  Unofficial JAX implementations of deep learning research papers
  ☆161Jun 25, 2022Updated 3 years ago
• f0uriest / quadax

  View on GitHub
  Numerical quadrature with JAX
  ☆75Jan 20, 2026Updated 3 weeks ago
• pnkraemer / probdiffeq

  View on GitHub
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆53Updated this week
• cgarciae / ciclo

  View on GitHub
  A functional training loops library for JAX
  ☆88Feb 13, 2024Updated 2 years ago
• alonfnt / bayex

  View on GitHub
  Minimal Implementation of Bayesian Optimization in JAX
  ☆102Apr 24, 2025Updated 9 months ago
• speed1313 / jaxkan

  View on GitHub
  JAX implementation of Kolmogorov Arnold Networks (KANs).
  ☆10May 7, 2024Updated last year
• f-dangel / einconv

  View on GitHub
  Convolutions and more as einsum for PyTorch
  ☆17Jun 6, 2024Updated last year
• cgarciae / nnx

  View on GitHub
  Neural Networks for JAX
  ☆84Sep 24, 2024Updated last year
• mpi4jax / mpi4jax

  View on GitHub
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆513Updated this week
• ucl-bug / jaxdf

  View on GitHub
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆132Updated this week
• rasbt / srgan-lightning-blog

  View on GitHub
  Sharing Deep Learning Research Models with Lightning Part 1: Building A Super Resolution App
  ☆16Jun 30, 2022Updated 3 years ago
• google-deepmind / chex

  View on GitHub
  ☆922Jan 29, 2026Updated 2 weeks ago
• cgarciae / treeo

  View on GitHub
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Feb 26, 2023Updated 2 years ago
• pnkraemer / tornadox

  View on GitHub
  Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…
  ☆21Jul 20, 2024Updated last year
• matthias-wright / jax-fid

  View on GitHub
  FID computation in Jax/Flax.
  ☆29Jul 17, 2024Updated last year
• n2cholas / awesome-jax

  View on GitHub
  JAX - A curated list of resources https://github.com/google/jax
  ☆2,041Jan 20, 2026Updated 3 weeks ago
• probml / dynamax

  View on GitHub
  A Python package for probabilistic state space modeling with JAX
  ☆924Jan 6, 2026Updated last month
• cgarciae / simple-pytree

  View on GitHub
  A dead simple Python package for creating custom JAX pytree objects
  ☆66Nov 14, 2024Updated last year
• jax-ml / jax-ai-stack

  View on GitHub
  ☆265Updated this week