Issues

• Mention how to use NNX metrics in the docs

  google/metrax#146 · Filter by author jeffcarp opened on May 5, 2026

• Make codebase compatible with ruff format

  google/metrax#140 · Filter by author jeffcarp opened on Jan 12, 2026
  1

• metrax.nnx metrics incompatible with nnx.MultiMetric due to strict kwargs in from_model_output

  google/metrax#132 · Filter by author steventango opened on Nov 22, 2025
  1

• FBetaScore missing merge method causes NotImplementedError with metrax.nnx wrapper

  google/metrax#131 · Filter by author steventango opened on Nov 22, 2025
  2

• Update dev dependencies to Python 3.13

  google/metrax#115 · Filter by author jeffcarp opened on Oct 24, 2025
  4

• Unpin JAX version after next jax_tpu_embedding release

  Task

  google/metrax#107 · Filter by author jeffcarp opened on Aug 14, 2025
  2

  

• Add from_logits flag to every Metrax metric

  good first issuehelp wanted
  google/metrax#105 · Filter by author jshin1394 opened on Aug 13, 2025
  4

  

• Failing Unit Tests

  google/metrax#101 · Filter by author Heidar-An opened on May 30, 2025
  2

• Makefile for local development

  google/metrax#96 · Filter by author Mrigankkh opened on May 18, 2025
  2

• Merge global test variables into a helper file

  google/metrax#89 · Filter by author jshin1394 opened on May 9, 2025

• Investigate if RecML can use Metrax metrics

  google/metrax#69 · Filter by author jeffcarp opened on Apr 23, 2025

• Add advanced tokenizer for LLM metrics

  google/metrax#68 · Filter by author jshin1394 opened on Apr 15, 2025

• Add Kernel Inception Distance (KID) metric

  good first issuehelp wanted
  google/metrax#65 · Filter by author jshin1394 opened on Apr 4, 2025
  2

  

• Add Total Variation (TV)

  good first issuehelp wanted
  google/metrax#63 · Filter by author jshin1394 opened on Apr 4, 2025
  3

• Add Mean Intersection over Union (mIoU) metric

  good first issuehelp wanted
  google/metrax#61 · Filter by author jshin1394 opened on Apr 4, 2025
  3

  

• Add FactorizedTopK metric

  google/metrax#20 · Filter by author jeffcarp opened on Mar 4, 2025
  

• Establish parity with TFRS

  google/metrax#19 · Filter by author jeffcarp opened on Mar 4, 2025

• Add recommendation for how to run distributed eval

  google/metrax#18 · Filter by author jeffcarp opened on Feb 27, 2025
  

• Search for custom subclassed CLU metrics in popular repos and send them PRs to upgrade

  good first issue
  google/metrax#3 · Filter by author jeffcarp opened on Feb 5, 2025
  1