grouped_mm_bf16
