pub fn initial_step_size(grad_norm: f64, max_step: Option<f64>) -> f64
Initialize step size for line search