scaled_logits = logits / temperature