Function embedding_avx2 

pub unsafe fn embedding_avx2(
    weight: &[f32],
    token_ids: &[u32],
    vocab_size: usize,
    dim: usize,
    output: &mut [f32],
)

AVX2 embedding lookup – delegates to scalar (memory-bound, no compute).

Safety

Requires AVX2 support.