Quantization-Aware Training (QAT) The QAT documentation has moved to the torchao docs: https://pytorch.org/ao/main/workflows/qat.html