Skip to content

[JAX] Integrate BF16 Grouped GEMM with on-device group sizes#2680

Open
jberchtold-nvidia wants to merge 16 commits intoNVIDIA:mainfrom
jberchtold-nvidia:gmm
Open

[JAX] Integrate BF16 Grouped GEMM with on-device group sizes#2680
jberchtold-nvidia wants to merge 16 commits intoNVIDIA:mainfrom
jberchtold-nvidia:gmm

Commits

Commits on Feb 24, 2026

Commits on Feb 25, 2026