Skip to content

[JAX] Grouped quant+GEMM custom partitioning rules#3058

Open
jberchtold-nvidia wants to merge 12 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/gmm-custom-partition-rules
Open

[JAX] Grouped quant+GEMM custom partitioning rules#3058
jberchtold-nvidia wants to merge 12 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/gmm-custom-partition-rules

Commits

Commits on May 26, 2026

Commits on Jun 1, 2026

Commits on Jun 2, 2026