Function embedding 

pub fn embedding(
    ids: &[u32],
    wte: &[f32],
    wpe: Option<&[f32]>,
    c: usize,
    pos: usize,
) -> Vec<f32>

Fused token + positional embedding gather.