PyTorch TorchAO Found out about this as I was optimizing model. https://docs.pytorch.org/ao/stable/index.html