pub unsafe extern "C" fn mlx_jvp(
res_0: *mut mlx_vector_array,
res_1: *mut mlx_vector_array,
fun: mlx_closure,
primals: mlx_vector_array,
tangents: mlx_vector_array,
) -> c_intpub unsafe extern "C" fn mlx_jvp(
res_0: *mut mlx_vector_array,
res_1: *mut mlx_vector_array,
fun: mlx_closure,
primals: mlx_vector_array,
tangents: mlx_vector_array,
) -> c_int