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

Add lm_head_kernel_transposed to remove the explicit-mode LM head tra…

6ff6d3c
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Sep 1, 2026 in 8s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

6ff6d3c Author: @NuojCheng <che******in​@google.com>

(Only the first commit for a unique contributor is listed.)