Function normalize_features 

pub fn normalize_features(
    data: &[Vec<f64>],
) -> (Vec<Vec<f64>>, Vec<f64>, Vec<f64>)

Normalize features to zero mean and unit variance. Returns (normalized_data, means, stds).