pub fn decode_batch_head(spec: &ModelSpec, dir: &str, bsz: usize) -> Authored
A batched decode step’s head alone: the last layer’s stacked output in, every sequence’s next token out.