https://github.com/tensorflow/model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
https://github.com/tensorflow/model-optimization

compression deep-learning keras machine-learning ml model-compression optimization pruning quantization quantized-networks quantized-neural-networks quantized-training sparsity tensorflow

Last synced: about 2 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/tensorflow/model-optimization
• Owner: tensorflow
• License: apache-2.0
• Created: 2018-10-31T20:34:28.000Z (almost 6 years ago)
• Default Branch: master
• Last Pushed: 2024-05-02T22:47:28.000Z (5 months ago)
• Last Synced: 2024-05-22T08:20:28.207Z (4 months ago)
• Topics: compression, deep-learning, keras, machine-learning, ml, model-compression, optimization, pruning, quantization, quantized-networks, quantized-neural-networks, quantized-training, sparsity, tensorflow
• Language: Python
• Homepage: https://www.tensorflow.org/model_optimization
• Size: 2.17 MB
• Stars: 1,472
• Watchers: 119
• Forks: 321
• Open Issues: 219
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE
  • Security: SECURITY.md

Awesome Lists containing this project

• awesome-llmops - TensorFlow Model Optimization - optimization.svg?style=flat-square) | (Optimizations / Profiling)
• awesome-deep-model-compression - TensorFlow Model Optimization Toolkit
• awesome-list - TensorFlow Model Optimization Toolkit - A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning (Deep Learning Framework / High-Level DL APIs)
• awesome-python-machine-learning-resources - GitHub - 48% open · ⏱️ 23.08.2022): (Tensorflow实用程序)

README

        
# TensorFlow Model Optimization Toolkit

The **TensorFlow Model Optimization Toolkit** is a suite of tools that users,

both novice and advanced, can use to optimize machine learning models for

deployment and execution.

Supported techniques include quantization and pruning for sparse weights.

There are APIs built specifically for Keras.

For an overview of this project and individual tools, the optimization gains,

and our roadmap refer to

[tensorflow.org/model_optimization](https://www.tensorflow.org/model_optimization).

The website also provides various tutorials and API docs.

The toolkit provides stable Python APIs.

## Installation

For installation instructions, see

[tensorflow.org/model_optimization/guide/install](https://www.tensorflow.org/model_optimization/guide/install).

## Contribution guidelines

**If you want to contribute to TensorFlow Model Optimization, be sure to review

the [contribution guidelines](CONTRIBUTING.md). This project adheres to

TensorFlow's

[code of conduct](https://github.com/tensorflow/tensorflow/blob/master/CODE_OF_CONDUCT.md).

By participating, you are expected to uphold this code.**

**We use

[GitHub issues](https://github.com/tensorflow/model-optimization/issues) for

tracking requests and bugs.**

## Maintainers

  

    Subpackage

    Maintainers

  

  

    tfmot.clustering

    Arm ML Tooling

  

  

    tfmot.quantization

    TensorFlow Model Optimization

  

  

    tfmot.sparsity

    TensorFlow Model Optimization

  

## Community

As part of TensorFlow, we're committed to fostering an open and welcoming

environment.

*   [TensorFlow Blog](https://blog.tensorflow.org): Stay up to date on content

    from the TensorFlow team and best articles from the community.