Function quantize_fp8 

pub fn quantize_fp8(
    data: &[f32],
    shape: &[usize],
    granularity: Granularity,
    variant: Fp8Variant,
) -> QuantizedFp8

Quantizes data (row-major, shape shape) to variant, scaling each group so its amax maps to the format’s largest finite value.