Main Page

Welcome to AI Model Efficiency Toolkit API Docs!

AI Model Efficiency Toolkit (AIMET) is a software toolkit that enables users to compress and quantize ML models. The resulting models returned by AIMET can be further trained (or fine-tuned) to dramatically improve accuracy lost due to quantization and compression.

AIMET is designed to work generically on any user-provided model. At present, AIMET supports TensorFlow and PyTorch training frameworks.

Please follow the links below to see AIMET APIs for either TensorFlow or PyTorch.

• AIMET APIs for PyTorch
  • PyTorch Model Guidelines
  • PyTorch Quantization API
  • PyTorch Model Compression API
  • PyTorch Model Visualization API for Compression
  • PyTorch Model Visualization API for Quantization

• AIMET APIs for TensorFlow
  • TensorFlow Quantization API
  • TensorFlow Model Compression API
  • TensorFlow Model Visualization API for Quantization
  • Using AIMET with Keras Model

Note

This documentation is auto-generated from the AIMET codebase using Sphinx

Indices and tables

• Index

• Module Index

• Search Page