Skip to content

[JAX] Hopper BF16 grouped GEMM v2 support#3083

Draft
jberchtold-nvidia wants to merge 3 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/hopper-bf16-gmm
Draft

[JAX] Hopper BF16 grouped GEMM v2 support#3083
jberchtold-nvidia wants to merge 3 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/hopper-bf16-gmm

Commits

Commits on Jun 4, 2026