Setup QAT training

[bglid]

Summary

Need to setup a QAT wrapper for the model to do another set of training with QAT enabled for evaluation comparison.

---

To-dos:

• Build QAT wrapper in PyTorch
• Move to HPC
• Run training and inference on this version
• Upload QAT models

---