lockwo/distreqx external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

lockwo/distreqx

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/lockwo/distreqx)

Close

lockwo / distreqxView on GitHubMore

• External links
• Readme badge

Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.

☆40Jan 16, 2026Updated 2 months ago

Alternatives and similar repositories for distreqx

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

• Artur-Galstyan / jaxonloader

  View on GitHub
  A dataloader, but for JAX
  ☆20May 17, 2024Updated last year
• danielward27 / flowjax

  View on GitHub
  ☆223Feb 21, 2026Updated last month
• JoeyTeng / jaxrenderer

  View on GitHub
  Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…
  ☆79Nov 14, 2023Updated 2 years ago
• flowersteam / sbmltoodejax

  View on GitHub
  ☆13Dec 20, 2024Updated last year
• clementpoiret / Equimo

  View on GitHub
  Implementation of Computer Vision Models in JAX (equinox)
  ☆20Jan 15, 2026Updated 2 months ago
• marin-community / haliax

  View on GitHub
  Named Tensors for Legible Deep Learning in JAX
  ☆217Nov 8, 2025Updated 4 months ago
• lasgroup / opax

  View on GitHub
  ☆19Jan 9, 2025Updated last year
• danielward27 / paramax

  View on GitHub
  A small library of paramaterizations and parameter constraints for PyTrees.
  ☆53Jan 23, 2026Updated last month
• yardenas / actsafe

  View on GitHub
  Scaling safe exploration to vision control
  ☆14Feb 19, 2025Updated last year
• speed1313 / jaxkan

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

  View on GitHub
  Matrix-free linear algebra in JAX.
  ☆162Dec 2, 2025Updated 3 months ago
• mancusolab / traceax

  View on GitHub
  Stochastic trace estimation using JAX
  ☆17Aug 20, 2025Updated 7 months ago
• matteoguarrera / awesome-equinox

  View on GitHub
  Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox
  ☆16Jun 24, 2024Updated last year
• jeremiecoullon / jax-tqdm

  View on GitHub
  Add a tqdm progress bar to your JAX scans and loops.
  ☆125May 9, 2025Updated 10 months ago
• nboyd / joltz

  View on GitHub
  JAX translation of boltz
  ☆25Aug 4, 2025Updated 7 months ago
• neuralgcm / coordax

  View on GitHub
  Coordinate Axes for JAX
  ☆62Feb 26, 2026Updated 3 weeks ago
• patrick-kidger / patdb

  View on GitHub
  A snappy + easy + pretty TUI debugger for Python.
  ☆69Oct 3, 2025Updated 5 months ago
• sparisi / cbet

  View on GitHub
  Change-Based Exploration Transfer
  ☆35Apr 24, 2022Updated 3 years ago
• RobertTLange / gymnax-blines

  View on GitHub
  Baselines for gymnax 🤖
  ☆75Apr 3, 2023Updated 2 years ago
• zishun / cuSolverRf-batch

  View on GitHub
  A complete example of batched refactorization in cuSOLVER.
  ☆13Dec 22, 2021Updated 4 years ago
• keraJLi / rejax

  View on GitHub
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆264Oct 31, 2025Updated 4 months ago
• tinker495 / Xtructure

  View on GitHub
  Xtructure is datastructure for using in JAX
  ☆21Mar 3, 2026Updated 2 weeks ago
• gdalle / sparsediffax

  View on GitHub
  Prototype for sparse automatic differentiation in JAX + Julia
  ☆21Jun 17, 2025Updated 9 months ago
• instadeepai / flashbax

  View on GitHub
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆274Sep 22, 2025Updated 6 months ago
• Howuhh / sac-n-jax

  View on GitHub
  Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
  ☆57May 21, 2023Updated 2 years ago
• akshatkaush / aurora-jax

  View on GitHub
  ☆27Aug 27, 2025Updated 6 months ago
• packquickly / schedule_free_optx

  View on GitHub
  Schedule free optimiser implemented in JAX using Optimistix
  ☆15May 29, 2024Updated last year
• google-deepmind / nao_top10

  View on GitHub
  ☆19Mar 1, 2023Updated 3 years ago
• j-towns / scanagram

  View on GitHub
  Tidy autoregressive inference in JAX
  ☆15Sep 1, 2025Updated 6 months ago
• FLAIROx / JaxGL

  View on GitHub
  Simple JAX Graphics Library.
  ☆37Nov 3, 2024Updated last year
• bolt-research / popgym-arcade

  View on GitHub
  Atari-style POMDPs
  ☆25Feb 13, 2026Updated last month
• kazewong / flowMC

  View on GitHub
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆256Oct 28, 2025Updated 4 months ago
• EdanToledo / Stoix

  View on GitHub
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆397Mar 1, 2026Updated 3 weeks ago
• ikostrikov / jaxrl2

  View on GitHub
  ☆53Jan 20, 2023Updated 3 years ago
• f0uriest / orthax

  View on GitHub
  Orthogonal polynomials with JAX
  ☆29Updated this week
• GalacticDynamics / coordinax

  View on GitHub
  Coordinates in JAX
  ☆29Mar 12, 2026Updated last week
• NIFTy-PPL / JAXbind

  View on GitHub
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆75Feb 12, 2026Updated last month
• alonfnt / pcax

  View on GitHub
  Differentiable Principal Component Analysis (PCA) implementation in JAX
  ☆35Apr 19, 2025Updated 11 months ago
• zombie-einstein / esquilax

  View on GitHub
  JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library
  ☆13Oct 12, 2025Updated 5 months ago