Function k2_ssd_bmm 

pub fn k2_ssd_bmm<B: Backend>(
    c_bnlgr: Tensor<B, 5>,
    b_bnlgr: Tensor<B, 5>,
) -> Tensor<B, 5>

Based on the Kernel 2 Triton reference _bmm_chunk_fwd_kernel (ssd_bmm.py).

Returns:

• cb_bngll used in K5.