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, Keras, and PyTorch training frameworks.

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

Note

AIMET Keras API requires Tensorflow 2.4 or later.

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

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

• AIMET APIs for Keras
  • Keras Model Quantization API

Note

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

Indices and tables

• Index

• Module Index

• Search Page