tensorflow/model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

What it solves

It provides a suite of tools to optimize machine learning models, making them more efficient for deployment and execution.

How it works

The toolkit uses techniques such as quantization and pruning for sparse weights to reduce model size and improve performance. It includes stable Python APIs and specific integrations for Keras.

Who it’s for

It is designed for both novice and advanced users who need to optimize their models for deployment.

Highlights

  • Quantization support
  • Pruning for sparse weights
  • Dedicated APIs for Keras
  • Stable Python APIs