Triton kernels for quantization