Function quantize 

pub fn quantize(w: &Array, group_size: i32, bits: i32) -> (Array, Array, Array)

w quantized along its last axis: (w_q, scales, biases); see Array::quantize.