aimet_torch API

aimet_torch

Important

aimet_torch package is upgraded to aimet_torch.v2 with more flexible, extensible, and PyTorch-friendly user interface!

aimet_torch 2 is fully backward compatible with all the public APIs of aimet_torch 1.x., please see Migrate to aimet_torch 2.

• aimet_torch.quantsim

• aimet_torch.nn

• aimet_torch.quantization

• aimet_torch.adaround

• aimet_torch.seq_mse

• aimet_torch.quantsim.config_utils

• aimet_torch.batch_norm_fold

• aimet_torch.cross_layer_equalization

• aimet_torch.model_preparer

• aimet_torch.model_validator

• aimet_torch.mixed_precision

• aimet_torch.quant_analyzer

• aimet_torch.autoquant

• aimet_torch.bn_reestimation

• aimet_torch.visualization_tools

• aimet_torch.layer_output_utils

• aimet_torch.peft

• aimet_torch.compress

aimet_torch.v1

If you still prefer to use aimet_torch 1.x, your imports should originate from the aimet_torch.v1 namespace.

• aimet_torch.v1.quantsim

• aimet_torch.v1.adaround

• aimet_torch.v1.seq_mse

• aimet_torch.v1.quant_analyzer

• aimet_torch.v1.autoquant

• aimet_torch.v1.amp