Skip to content

Remove the explicit-sharding LM head transpose via lm_head_kernel_transposed - #5091

Draft
NuojCheng wants to merge 1 commit into
mainfrom
chengnuojin-explicit-lmhead-orientation
Draft

Remove the explicit-sharding LM head transpose via lm_head_kernel_transposed#5091
NuojCheng wants to merge 1 commit into
mainfrom
chengnuojin-explicit-lmhead-orientation

Commits

Commits on Sep 1, 2026