Function jvp 

pub fn jvp<F>(
    f: F,
    primals: &[Tensor],
    tangents: &[Option<Tensor>],
) -> Result<JvpResult>
where
    F: FnOnce(&[Tensor]) -> Result<Vec<Tensor>>,