- Notifications
You must be signed in to change notification settings - Fork0
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
License
sarvex/tensorflow-model-optimization
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
TheTensorFlow Model Optimization Toolkit is a suite of tools that users,both novice and advanced, can use to optimize machine learning models fordeployment 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 totensorflow.org/model_optimization.The website also provides various tutorials and API docs.
The toolkit provides stable Python APIs.
For installation instructions, seetensorflow.org/model_optimization/guide/install.
If you want to contribute to TensorFlow Model Optimization, be sure to reviewthecontribution guidelines. This project adheres toTensorFlow'scode of conduct.By participating, you are expected to uphold this code.
We useGitHub issues fortracking requests and bugs.
| Subpackage | Maintainers |
|---|---|
| tfmot.clustering | Arm ML Tooling |
| tfmot.quantization | TensorFlow Model Optimization |
| tfmot.sparsity | TensorFlow Model Optimization |
As part of TensorFlow, we're committed to fostering an open and welcomingenvironment.
- TensorFlow Blog: Stay up to date on contentfrom the TensorFlow team and best articles from the community.
About
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
Resources
License
Code of conduct
Contributing
Security policy
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Packages0
Uh oh!
There was an error while loading.Please reload this page.
Languages
- Python82.2%
- Jupyter Notebook13.1%
- Starlark4.3%
- Other0.4%