Function mlx_closure_custom_jvp_apply 

pub unsafe extern "C" fn mlx_closure_custom_jvp_apply(
    res: *mut mlx_vector_array,
    cls: mlx_closure_custom_jvp,
    input_0: mlx_vector_array,
    input_1: mlx_vector_array,
    input_2: *const c_int,
    input_2_num: usize,
) -> c_int