pub fn head(spec: &ModelSpec, dir: &str) -> Authored
The final norm and LM head alone, reading the last layer’s output.