tensorflow/model-optimization

View on GitHub

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

Stars: 1.6k
Forks: 349
Open beginner issues: 1
Indexed issues: 201
Avg merge: 3d 2h
Merged PRs (30d): 1

Dominant language: Python
License: Apache-2.0
Last GitHub push: Aug 24, 2026
Latest indexed: Sep 19, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed

Beginner friendly Everything

Open All

201 open issues indexed

Support for nested `tf.keras.layers.Layer` Open

feature request

tensorflow/model-optimization#638 · 5 comments · 2 reactions · 1 assignee · Opened Feb 26, 2021
Symmetric quantization returns nonzero value for 0 on GPU Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 45/100

tensorflow/model-optimization#635 · 1 comment · Opened Feb 22, 2021
block_size for pruning 3D weights Open

feature request

tensorflow/model-optimization#634 · 1 comment · 2 reactions · 1 assignee · Opened Feb 22, 2021
OOM errors when running tests with bazel Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

tensorflow/model-optimization#633 · Opened Feb 19, 2021
How to get quantized weights from QAT model? Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

tensorflow/model-optimization#626 · 11 comments · Opened Feb 3, 2021
After QAT and TFLite converter, the type of input and output of averagepooling2d node is not same Open

bug

tensorflow/model-optimization#622 · 7 comments · 4 assignees · Opened Jan 25, 2021
Is it possible to implement post-training pruning? Open

feature request

tensorflow/model-optimization#621 · 4 comments · 1 reaction · 1 assignee · Opened Jan 20, 2021
Pruning does not reduce inference time. Open

feature request

Difficulty 5/5 Over a week Newbie friendliness 20/100

tensorflow/model-optimization#611 · 3 comments · Opened Jan 8, 2021
TFJS Support Open

feature request

Difficulty 5/5 Over a week Newbie friendliness 25/100

tensorflow/model-optimization#606 · Opened Dec 20, 2020
Cannot restore a checkpoint on a pruned model without 'Unresolved object in checkpoint: (root).optimizer.*' Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 25/100

tensorflow/model-optimization#603 · 2 comments · Opened Dec 3, 2020
layer is not supported for Quantization aware training Open

feature request

Difficulty 5/5 Over a week Newbie friendliness 25/100

tensorflow/model-optimization#602 · 2 comments · Opened Nov 30, 2020
Polynomial decay for pruning schedule doesn't prune the weight Open

bug

tensorflow/model-optimization#601 · 1 comment · 1 assignee · Opened Nov 25, 2020
Unable to get tfmot version from API Open

bug

tensorflow/model-optimization#594 · 2 comments · 1 reaction · 1 assignee · Opened Nov 17, 2020
Missing custom input quantization Open

feature request

Difficulty 4/5 3-5 days Newbie friendliness 35/100

tensorflow/model-optimization#589 · 2 comments · Opened Nov 7, 2020
pruning model from model zoo Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

tensorflow/model-optimization#572 · 1 comment · Opened Oct 14, 2020
support all number types in BitpackingEncodingStage Open

feature request

Difficulty 4/5 3-5 days Newbie friendliness 25/100

tensorflow/model-optimization#560 · Opened Sep 30, 2020
AdaptiveEncodingStageInterface decode should be able to receive and update a state Open

feature request

Difficulty 5/5 Over a week Newbie friendliness 35/100

tensorflow/model-optimization#558 · Opened Sep 27, 2020
AttributeError: 'NoOpActivation' object has no attribute '__name__' Open

bug

Difficulty 3/5 1-2 days Newbie friendliness 35/100

tensorflow/model-optimization#556 · 3 comments · Opened Sep 25, 2020
Applying constraint on (Conv+BN+ReLU) folded weights Open

question technique:qat

tensorflow/model-optimization#549 · 3 comments · 1 assignee · Opened Sep 18, 2020
quantization.keras.quantize_model function runtime error with mobilenet v3 Open

bug

tensorflow/model-optimization#546 · 6 comments · 1 reaction · 1 assignee · Opened Sep 15, 2020
Support for block sparse matrices Open

feature request

tensorflow/model-optimization#542 · 2 comments · 1 assignee · Opened Sep 13, 2020
Make TFMOT model files available Open

feature request technique:all

Difficulty 5/5 Over a week Newbie friendliness 25/100

tensorflow/model-optimization#529 · Opened Sep 1, 2020
Quantized wrapped layer non-trainable Open

feature request technique:qat

tensorflow/model-optimization#527 · 1 comment · 1 assignee · Opened Aug 24, 2020
How to preform progressive quantization aware training using this framework? Open

question

Difficulty 5/5 Over a week Newbie friendliness 20/100

tensorflow/model-optimization#525 · 1 comment · Opened Aug 20, 2020
Mismatch in number of weights when loading quantized model (activation layer) Open

bug

tensorflow/model-optimization#524 · 9 comments · 1 assignee · Opened Aug 20, 2020
CONV2D, DENSE layer with default QAT can't generate correct int8 quantized nodes in TFLite model Open

bug

Difficulty 4/5 3-5 days Newbie friendliness 35/100

tensorflow/model-optimization#523 · 4 comments · Opened Aug 19, 2020
Optimizing models that use TensorFlow Addons activations, layers, etc Open

feature request technique:all

tensorflow/model-optimization#471 · 3 comments · 1 reaction · 1 assignee · Opened Jul 21, 2020
Post quantization does not utilize GPU Open

feature request

Difficulty 5/5 Over a week Newbie friendliness 25/100

tensorflow/model-optimization#454 · 3 comments · Opened Jul 8, 2020
How to user quantize to imporve inference performance on tensorflow-serving? Open

bug technique:qat

tensorflow/model-optimization#450 · 3 comments · 2 reactions · 1 assignee · Opened Jul 2, 2020
[Feature request or potential bug] Override of default default_8bit_quantize_layout_transform Open

feature request technique:qat

tensorflow/model-optimization#445 · 10 comments · 2 assignees · Opened Jun 30, 2020
Issue with loading quantization aware trained model Open

bug

tensorflow/model-optimization#426 · 13 comments · 1 reaction · 1 assignee · Opened Jun 12, 2020
sparsity.prune_low_magnitude fails with mixed precision policy mixed_float16 Open

feature request technique:pruning

tensorflow/model-optimization#409 · 6 comments · 1 assignee · Opened May 28, 2020
Behaviour of stripped models and sequence masking Open

bug technique:pruning

Difficulty 4/5 3-5 days Newbie friendliness 25/100

tensorflow/model-optimization#403 · 4 comments · Opened May 21, 2020
H5 to Pb Conversion with Fake Quantization Node Fails Open

bug technique:qat

Difficulty 4/5 3-5 days Newbie friendliness 35/100

tensorflow/model-optimization#381 · 1 comment · 1 reaction · Opened May 11, 2020
QAT (quantization aware training) Support quantizing models recursively Open

feature request technique:qat

tensorflow/model-optimization#377 · 24 comments · 1 assignee · Opened May 4, 2020
Tracking TFMOT 1.0.0 Release Open

technique:all

Difficulty 5/5 Over a week Newbie friendliness 15/100

tensorflow/model-optimization#371 · Opened Apr 22, 2020
quantize_model and tf.keras.mixed_precision.experimental.Policy("mixed_float16") throws ValueError Open

bug technique:qat

Difficulty 4/5 3-5 days Newbie friendliness 35/100

tensorflow/model-optimization#367 · 2 comments · 2 reactions · Opened Apr 18, 2020
Support Dense+BatchNorm in QAT Open

bug technique:qat

tensorflow/model-optimization#363 · 6 comments · 4 reactions · 1 assignee · Opened Apr 16, 2020
Support for Conv1D Quantization Open

feature request technique:qat

tensorflow/model-optimization#362 · 32 comments · 5 reactions · 1 assignee · Opened Apr 15, 2020
Move TFMOT Library Code Off Private APIs Open

technique:all

Difficulty 4/5 3-5 days Newbie friendliness 35/100

tensorflow/model-optimization#305 · 2 comments · Opened Mar 18, 2020
Pruning: improve custom training loop API Open

feature request priority:low technique:pruning

tensorflow/model-optimization#271 · 5 comments · 1 reaction · 1 assignee · Opened Feb 27, 2020
Pruning fails for siamese networks Open

feature request priority:low technique:pruning

tensorflow/model-optimization#270 · 4 comments · 1 assignee · Opened Feb 26, 2020
Pruning: Checkpointed Models Not Guaranteed To Be Sparse Open

contributions welcome feature request technique:pruning

tensorflow/model-optimization#206 · 5 comments · 1 assignee · Opened Dec 30, 2019
Sparsity Runtime Integration with TF/TFLite for Latency Improvements Open

feature request technique:pruning

tensorflow/model-optimization#173 · 30 comments · 21 reactions · 1 assignee · Opened Dec 6, 2019
Pruning: Keras subclassed model increased support Open

contributions welcome feature request technique:pruning

tensorflow/model-optimization#155 · 7 comments · 12 reactions · 1 assignee · Opened Nov 11, 2019
Opensource Mobilenet V1 Image Classification Training Script Open

feature request

tensorflow/model-optimization#133 · 5 comments · 2 assignees · Opened Oct 23, 2019
Object detection API Open

technique:pruning

tensorflow/model-optimization#132 · 12 comments · 1 assignee · Opened Oct 22, 2019
Make Model Optimization Project Contributor Friendly Open

feature request technique:all

tensorflow/model-optimization#131 · 3 comments · 6 reactions · 1 assignee · Opened Oct 21, 2019
raise FailedPreconditionError while training model Open

bug technique:pruning

Difficulty 4/5 3-5 days Newbie friendliness 20/100

tensorflow/model-optimization#46 · 4 comments · Opened Jul 29, 2019
Does anyone try to apply this project to BERT? Open

technique:pruning

tensorflow/model-optimization#29 · 7 comments · 2 assignees · Opened Jul 5, 2019

tensorflow/model-optimization

Get new issues in your inbox