BLOCK_M ======= Auto-generated from mlx_code_defs .. raw:: html
torch.nn.attention.flex_attention.FlexKernelOptions.BLOCK_M
mlx.core.block_masked_mm