Function mlp 

fn mlp(
    gate: &QLinear<'_>,
    up: &QLinear<'_>,
    down: &QLinear<'_>,
    x: &[f32],
) -> Vec<f32>