[JAX] Add bias support for v2 grouped GEMM path#2744

Closed

jberchtold-nvidia wants to merge 1 commit intoNVIDIA:mainfrom

jberchtold-nvidia:jberchtold/gmm-bias

Commits on Mar 6, 2026

[JAX] Add bias support for v2 grouped GEMM path
jberchtold-nvidia
committed