Function decode_batch_embed 

pub fn decode_batch_embed(
    spec: &ModelSpec,
    dir: &str,
    toks: &[usize],
) -> Authored

A batched decode step’s embedding lookups alone: the stacked X out.