Repository navigation

DLRMv4: compute the last HSTU layer only for the rows the loss reads - #903

Merged

ShriyaRishab merged 1 commit into

mlcommons:masterfrom

jiaweche:targets-only

Sep 20, 2026

Merged

DLRMv4: compute the last HSTU layer only for the rows the loss reads#903
ShriyaRishab merged 1 commit into
mlcommons:masterfrom
jiaweche:targets-only

Commits on Aug 28, 2026
- [DLRMv4] Shrink the last HSTU layer to the rows the loss reads
  jiaweche
  committed
  e60300b