Skip to content

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

Open
jberchtold-nvidia wants to merge 27 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 27 commits intoNVIDIA:mainfrom
jberchtold-nvidia:jberchtold/gmm-refactor

Commits

Commits on Mar 9, 2026

Commits on Mar 10, 2026

Commits on Mar 11, 2026

Commits on Mar 12, 2026

Commits on Mar 13, 2026

Commits on Mar 17, 2026

Commits on Mar 23, 2026

Commits on Mar 24, 2026

Commits on Mar 25, 2026