Repository navigation

[Metal] Add staged affine_qmm_t_nax kernels with split-k - #4682

Merged

zcbenz merged 4 commits into

ml-explore:mainfrom

wyanzhao:qmm-nax-mid-m

Oct 20, 2026

Merged

[Metal] Add staged affine_qmm_t_nax kernels with split-k#4682
zcbenz merged 4 commits into
ml-explore:mainfrom
wyanzhao:qmm-nax-mid-m

Commits on Oct 20, 2026
- Route mid-size M quantized matmuls through a split-K NAX qmm
  
  wyanzhao
  authored and
  zcbenz
  committed
  bffb483
- Faster transposed affine NAX qmm: 16-row simdgroups, double-buffered weights
  
  wyanzhao
  authored and
  zcbenz
  committed
  930bacf
- Address NAX quantized matmul review feedback
  
  wyanzhao
  authored and
  zcbenz
  committed
  804d7cd
- nit
  zcbenz
  committed
  93f103c