Update QAT: add grad clipping, torch.compile, collate fn

Update QAT: add grad clipping, torch.compile, collate fn #5667