Function average_host_grads 

pub fn average_host_grads(
    per_rank: &[Vec<Vec<f32>>],
) -> Result<Vec<Vec<f32>>, Box<dyn Error>>

Average host-side parameter vectors (unit-testable without GPU).