Function compute_gradient_stats 

pub fn compute_gradient_stats(
    gradients: &HashMap<String, Array1<f64>>,
) -> Vec<GradientStats>

Compute gradient statistics for all layers in a gradient dictionary.