tensorflow/model-optimization

Health v2 · maintenance only

Adoption not part of the score

1578 stars · 346 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

The TensorFlow Model Optimization Toolkit (tfmot) is a Python library providing tools to optimize machine learning models for deployment, including quantization, pruning for sparsity, and weight clustering, with APIs built for Keras. It helps reduce model latency, size, and power consumption for cloud and edge inference.

Use cases

• quantize a trained Keras model for TensorFlow Lite deployment
• prune neural network weights to increase sparsity and shrink model size
• reduce model latency and memory footprint on mobile and IoT devices
• cluster model weights to improve compression for deployment
• compress a deep learning model to fit edge device storage limits
• speed up inference cost for models served in the cloud

When to choose

• you deploy TensorFlow or Keras models to mobile, IoT, or other edge devices with tight memory, power, or latency budgets
• you want post-training or quantization-aware training with stable Keras APIs
• you need smaller model files for download or on-device storage
• you target hardware accelerators that benefit from quantized or sparse weights

When to avoid

• your models are built with PyTorch, JAX, or other non-TensorFlow frameworks
• you need general-purpose model compression outside the TensorFlow ecosystem
• you require cutting-edge optimization techniques not covered by quantization, pruning, or clustering
• your models run only on servers with no resource constraints

Facets

library · maturity stable

machine-learning deep-learning llm-training machine-learning deep-learning mobile-development iot performance python cross-platform quantization pruning sparsity model-compression keras tensorflow edge-deployment model-optimization

Member repositories

For agents

markdown · JSON · MCP: product_card(name="tensorflow/model-optimization")