Function prediction_chunk_rows 

pub fn prediction_chunk_rows(
    parameter_dim: usize,
    local_dim: usize,
    total_rows: usize,
) -> usize

Select the row count for prediction-time covariance work.

A prediction row can keep roughly four parameter_dim × local_dim f64 workspaces live while gradients and covariance solves are assembled. This central policy prevents predictors from drifting to different memory budgets and chunk bounds for the same operation.