Function diagonal_preconditioner 

pub fn diagonal_preconditioner(a: &CsrMatrix) -> Vec<f64>

Build a diagonal (Jacobi) preconditioner: returns 1 / a_ii for each i.

Returns 1.0 where the diagonal is zero (to avoid division by zero).