cgarciae/jax_metrics 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

cgarciae/jax_metrics

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/cgarciae/jax_metrics)

Close

cgarciae / jax_metricsView on GitHubMore

• External links
• Readme badge

A metrics library for the JAX ecosystem

☆40Mar 16, 2023Updated 2 years ago

Alternatives and similar repositories for jax_metrics

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

• cgarciae / ciclo

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

  View on GitHub
  Tools for JAX
  ☆50Mar 3, 2026Updated last week
• cgarciae / nnx

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

  View on GitHub
  Pytrees + dataclasses ❤️
  ☆75Dec 19, 2025Updated 2 months ago
• cgarciae / jax-gravity

  View on GitHub
  ☆24Jul 31, 2023Updated 2 years ago
• google / tree-math

  View on GitHub
  Mathematical operations for JAX pytrees
  ☆209Dec 5, 2024Updated last year
• optframe / pyoptframe

  View on GitHub
  Python bindings for OptFrame C++ Functional Core
  ☆13May 18, 2025Updated 9 months ago
• dirmeier / ramsey

  View on GitHub
  Probabilistic deep learning using JAX
  ☆15Feb 8, 2025Updated last year
• google / jax-recommenders

  View on GitHub
  ☆11Oct 29, 2022Updated 3 years ago
• deterministic-algorithms-lab / Jax-Journey

  View on GitHub
  A pathway and collection of resources to learning Jax from beginning to advance.
  ☆11Jan 2, 2021Updated 5 years ago
• NeilGirdhar / efax

  View on GitHub
  Exponential families for JAX
  ☆76Mar 3, 2026Updated last week
• google-deepmind / distrax

  View on GitHub
  ☆624Jan 23, 2026Updated last month
• n2cholas / jax-resnet

  View on GitHub
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆119Jun 5, 2022Updated 3 years ago
• lzhangbv / eva

  View on GitHub
  [ICLR 2023] Eva: Practical Second-order Optimization with Kronecker-vectorized Approximation
  ☆12Jul 31, 2023Updated 2 years ago
• jax-ml / oryx

  View on GitHub
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆302Feb 27, 2026Updated last week
• borisdayma / clip-jax

  View on GitHub
  Train vision models using JAX and 🤗 transformers
  ☆100Dec 14, 2025Updated 2 months ago
• gclouduniverse / reproducibility-deprecated

  View on GitHub
  ☆16Mar 13, 2025Updated 11 months ago
• DifferentiableUniverseInitiative / jax-cosmo-paper

  View on GitHub
  Paper introducing jax-cosmo
  ☆13Apr 27, 2023Updated 2 years ago
• rdyro / torch2jax

  View on GitHub
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Feb 15, 2026Updated 3 weeks ago
• privong / pymccorrelation

  View on GitHub
  Correlation coefficients with uncertainties
  ☆12Nov 12, 2025Updated 3 months ago
• google / CommonLoopUtils

  View on GitHub
  CLU lets you write beautiful training loops in JAX.
  ☆367Mar 3, 2026Updated last week
• cgarciae / treeo

  View on GitHub
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Feb 26, 2023Updated 3 years ago
• probml / pgm-jax

  View on GitHub
  Probabilistic Graphical Models in JAX
  ☆16Oct 11, 2022Updated 3 years ago
• coax-dev / coax

  View on GitHub
  Modular framework for Reinforcement Learning in python
  ☆183Feb 1, 2023Updated 3 years ago
• google-deepmind / jmp

  View on GitHub
  JMP is a Mixed Precision library for JAX.
  ☆211Jan 30, 2025Updated last year
• poets-ai / elegy

  View on GitHub
  A High Level API for Deep Learning in JAX
  ☆476Dec 15, 2022Updated 3 years ago
• jax-ml / jax-ai-stack

  View on GitHub
  ☆273Mar 2, 2026Updated last week
• robot-acceleration / RBDReference

  View on GitHub
  A Python reference implementation of rigid body dynamics algorithms
  ☆18Jul 17, 2023Updated 2 years ago
• martiningram / jax_advi

  View on GitHub
  ☆16Apr 8, 2025Updated 11 months ago
• cgarciae / treex

  View on GitHub
  A Pytree Module system for Deep Learning in JAX
  ☆213Feb 26, 2023Updated 3 years ago
• xl0 / lovely-jax

  View on GitHub
  JAX Arrays for human consumption
  ☆110Oct 23, 2025Updated 4 months ago
• probml / dynamax

  View on GitHub
  A Python package for probabilistic state space modeling with JAX
  ☆935Jan 6, 2026Updated 2 months ago
• khdlr / augmax

  View on GitHub
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆42May 16, 2025Updated 9 months ago
• anyscale / rl-course

  View on GitHub
  ☆25Jan 2, 2023Updated 3 years ago
• RockySJ / ampo

  View on GitHub
  ☆15Oct 20, 2020Updated 5 years ago
• adonath / gmmx

  View on GitHub
  Gaussian Mixture Models in Jax
  ☆24Jul 14, 2025Updated 7 months ago
• dfm / jpu

  View on GitHub
  JAX + Units
  ☆52Mar 2, 2026Updated last week
• juliuskunze / cwvae-jax

  View on GitHub
  Clockwork VAEs in JAX/Flax
  ☆32Jul 16, 2021Updated 4 years ago
• google / praxis

  View on GitHub
  ☆193Feb 27, 2026Updated last week