Function kimi_forward_token 

pub fn kimi_forward_token(
    weights: &KimiDecoderWeights,
    cfg: &KimiDecoderConfig,
    mla_cfg: &MlaConfig,
    kda_cfg: &KdaConfig,
    token_id: usize,
    state: &mut KimiDecodeState,
) -> Vec<f32>

One decode step across every layer.