Implement the AO API in torchchat quantization handlers and unify logic.

Implement the AO API in torchchat quantization handlers and unify logic. #1866