Skip to content

[JAX] Grouped GEMM Refactor to use first_dims and last_dims#2749

Open
jberchtold-nvidia wants to merge 14 commits intoNVIDIA:mainfrom
jberchtold-nvidia:jberchtold/gmm-refactor
Open

[JAX] Grouped GEMM Refactor to use first_dims and last_dims#2749
jberchtold-nvidia wants to merge 14 commits intoNVIDIA:mainfrom
jberchtold-nvidia:jberchtold/gmm-refactor

Commits

Commits on Mar 9, 2026

Commits on Mar 10, 2026

Commits on Mar 11, 2026