Function log_sum_exp 

pub fn log_sum_exp(logits: &[f32]) -> f64

Numerically-stable log-sum-exp over logits (f64 accumulation).