pub unsafe extern "C" fn mlx_random_truncated_normal( res: *mut mlx_array, lower: mlx_array, upper: mlx_array, shape: *const c_int, shape_num: usize, dtype: mlx_dtype, key: mlx_array, s: mlx_stream, ) -> c_int