Skip to content

[JAX] Fix GEMM partitioning to reduce over nested contracting mesh axes - #3429

Merged
phu0ngng merged 4 commits into
NVIDIA:mainfrom
phu0ngng:jax/fix_gemm_partitioning
Aug 28, 2026
Merged

[JAX] Fix GEMM partitioning to reduce over nested contracting mesh axes#3429
phu0ngng merged 4 commits into
NVIDIA:mainfrom
phu0ngng:jax/fix_gemm_partitioning

Merge branch 'main' into jax/fix_gemm_partitioning

dda83f8
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 28, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO