Function mlx_detail_vmap_trace 

pub unsafe extern "C" fn mlx_detail_vmap_trace(
    res_0: *mut mlx_vector_array,
    res_1: *mut mlx_vector_array,
    fun: mlx_closure,
    inputs: mlx_vector_array,
    in_axes: *const c_int,
    in_axes_num: usize,
) -> c_int