Skip to content

Avoid zero-initializing JAX grouped GEMM outputs - #592

Closed
mgoldfarb-nvidia wants to merge 1 commit into
NVIDIA:developfrom
mgoldfarb-nvidia:mgoldfarb/jax-grouped-gemm-output-init
Closed

Avoid zero-initializing JAX grouped GEMM outputs#592
mgoldfarb-nvidia wants to merge 1 commit into
NVIDIA:developfrom
mgoldfarb-nvidia:mgoldfarb/jax-grouped-gemm-output-init

Commits

Commits on Aug 14, 2026