Home/Model Training/model-optimization

model-optimization

tensorflow/model-optimization

Toolkit for optimizing ML models in Keras and TensorFlow

GraphCanon updated Oct 3, 2026 · GitHub synced Oct 3, 2026

45views this month

1.6k stars348 forksLast push Oct 2, 2026 Python Apache-2.0

Decision brief

Toolkit for optimizing ML models in Keras and TensorFlow, focusing on quantization and pruning.

Good fit when

When you are working with Keras or TensorFlow models and need to apply post-training quantization or pruning techniques to minimize model size and enhance inference speed.
If your project requires compression that is explicitly supported by the toolkit for better performance on edge devices.

Avoid when

Do not use this toolkit if you are working with ML models outside of Keras and TensorFlow frameworks, as it does not support other popular frameworks like PyTorch.
Avoid using this toolkit when detailed customization is needed beyond its quantization and pruning options, since the available methods might be too limited for complex optimization tasks.

Observed Jul 16, 2026 · Source: enrich:decision_facts

model-optimization alternatives Trust report

Verify the decision

Maintenance and security

Full trust report

Maintenance: Very active (1d since push); As of Oct 3, 2026
Provenance: Not a fork · Organization account; As of Oct 3, 2026
Security (OSV): No criticals; As of Jul 11, 2026

Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.

Install

pip install model-optimization

How it fits your stack(2)

Typed graph edges - alternatives, integrations, successors, and dependencies. Ranked by relationship type, not raw GitHub stars.

Integrates

TensorFlow Model Optimization Toolkit assists in optimizing models created within TensorFlow, addressing aspects like performance optimization for deployment and execution.

TensorFlow Model Optimization can optimize models created with various frameworks, including PyTorch which is used by Ultralytics, making them compatible for inference optimization.

Relationship graph

Optional deeper exploration of typed edges and category neighbours.

Similar tools

Same-category neighbours not already linked as typed edges.

Hardware-aware model recommendation tool

Visualizer for neural network, deep learning and machine learning models

Easily fine-tune, evaluate and deploy open source LLMs/VLMs

Evidence and technical details

Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.

Overview

Provides functionalities including quantization and pruning to optimize models for deployment.

Capability facts

Languages: python
Source: github.language · Oct 3, 2026

Categories

Tags

compression deep-learning keras machine-learning ml model-compression optimization pruning

README

Installation For installation instructions, see tensorflow.org/model optimization/guide/install.

For agents

This page has a .md twin and JSON over the API.

/tools/tensorflow-model-optimization.md GET /api/graphcanon/tools/tensorflow-model-optimization GET /api/graphcanon/graph?tool=tensorflow-model-optimization

Was this helpful?

Anonymous feedback helps us improve pages and translations.