Function softmax_init_state 

pub fn softmax_init_state<E: Float>(
    num_rows_per_unit: u32,
) -> (RowWise<E>, RowWise<E>)

Initial running state (m, l) for the online softmax over a single tile row.