DeepSpeed

381e61b3 - Move weight quantization into separate function in base container