456 add support for jax 08 traces #501

Merged

gabeweisz merged 25 commits intomainfrom

456-add-support-for-traces-generated-by-jax-08

Mar 29, 2026

Commits on Feb 20, 2026

Migrate from tensorboard-plugin-profile to xprof for JAX trace processing (#456)

gphuang
and
cursoragent
committed
Fix xprof compatibility for HLO metadata parsing and operand resolution

gphuang
and
cursoragent
committed

Commits on Feb 24, 2026

Add xprof to setup.py, improve DataLoader error on conversion failure

gphuang
and
cursoragent
committed
Rename JAX conv traces for clarity (Option A)

gphuang
and
cursoragent
committed
Apply Black formatting to fix lint

gphuang
and
cursoragent
committed
Address Copilot PR review comments

gphuang
and
cursoragent
committed
Pin Black to 26.1.0 in lint workflow for consistent formatting

gphuang
and
cursoragent
committed
Ignore process_name, process_label, thread_name in compare test

gphuang
and
cursoragent
committed

Commits on Feb 25, 2026

Commits on Feb 26, 2026

Commits on Feb 27, 2026

Merge branch 'main' into 456-add-support-for-traces-generated-by-jax-08
gphuang
authored

Commits on Mar 2, 2026

Apply Black formatting to tree_perf.py
gphuang
committed

Commits on Mar 9, 2026

Merge branch 'main' into 456-add-support-for-traces-generated-by-jax-08
gphuang
authored

Commits on Mar 23, 2026

Merge branch 'main' into 456-add-support-for-traces-generated-by-jax-08
gphuang
authored

Commits on Mar 25, 2026

Create performance model for reduction operators (#545)

gabeweisz
and
Copilot
authored

Commits on Mar 27, 2026

Merge origin/main: adopt CSV tests and simplified deterministic sort
gphuang
committed