pub unsafe extern "C" fn mlx_take( res: *mut mlx_array, a: mlx_array, indices: mlx_array, s: mlx_stream, ) -> c_int