Function compute_gradient_fwd_k 

pub fn compute_gradient_fwd_k<const K: usize, F>(
    inputs: &[f64],
    func: F,
) -> (f64, Vec<f64>)
where
    F: Fn(&[JetK<f64, K>]) -> JetK<f64, K>,