Function mlx_trace_axes 

pub unsafe extern "C" fn mlx_trace_axes(
    res: *mut mlx_array,
    a: mlx_array,
    offset: c_int,
    axis1: c_int,
    axis2: c_int,
    dtype: mlx_dtype,
    s: mlx_stream,
) -> c_int