AIMET PyTorch Quantization APIs

In order to make full use of AIMET Quantization features, there are several guidelines users are encouraged to follow when defining PyTorch models. AIMET provides APIs which can automate some of the model definition changes and checks whether AIMET Quantization features can be applied on PyTorch model.

Users should first invoke Model Preparer API before using any of the AIMET Quantization features.
AIMET Quantization for PyTorch Models provides the following functionality.
If a user wants to use Multi-GPU with CLE or QAT, they can refer to: