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

Avoid zero-initializing JAX grouped GEMM outputs

3da9256
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs