pub fn argmax(logits: &[f32]) -> u32
The lowest-id argmax of logits (greedy pick after penalties/bias, when temperature == 0).
logits
temperature == 0