pub unsafe extern "C" fn mlx_conv2d(
res: *mut mlx_array,
input: mlx_array,
weight: mlx_array,
stride_0: c_int,
stride_1: c_int,
padding_0: c_int,
padding_1: c_int,
dilation_0: c_int,
dilation_1: c_int,
groups: c_int,
s: mlx_stream,
) -> c_int