sparseml
Model optimizer
Enables the creation of smaller neural network models through efficient pruning and quantization techniques
Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models
2k stars
49 watching
148 forks
Language: Python
last commit: 4 months ago
Linked from 1 awesome list
automlcomputer-vision-algorithmsdeep-learning-algorithmsdeep-learning-librarydeep-learning-modelsimage-classificationkerasnlpobject-detectiononnxpruningpruning-algorithmspytorchsmaller-modelssparsificationsparsification-recipessparsitytensorflowtransfer-learning
ethicalml/awesome-production-machine-learningRelated projects:
| Repository | Description | Stars |
|---|---|---|
 intel/neural-compressor | Tools and techniques for optimizing large language models on various frameworks and hardware platforms. | 2,226 |
 hiroyuki-kasai/sparsegdlibrary | A collection of unconstrained optimization algorithms for sparse modeling in MATLAB | 53 |
 minimaxir/automl-gs | Automates machine learning model creation and optimization for complex datasets | 1,853 |
 microsoft/archai | Automates the search for optimal neural network configurations in deep learning applications | 467 |
 lge-arc-advancedai/auptimizer | Automates model building and deployment process by optimizing hyperparameters and compressing models for edge computing. | 200 |
 google-research/sputnik | A library of optimized GPU kernels for sparse matrix operations used in deep learning. | 249 |
 xternalz/sdpoint | A deep learning method for optimizing convolutional neural networks by reducing computational cost while improving regularization and inference efficiency. | 18 |
 dmlc/mxnet-memonger | A tool for optimizing deep learning models to reduce memory usage without sacrificing performance | 308 |
 brml/climin | A framework for optimizing machine learning functions using gradient-based optimization methods. | 180 |
 herilalaina/mosaic_ml | Automated machine learning with tree search optimization | 16 |
 balavenkatesh3322/nlp-pretrained-model | A collection of pre-trained natural language processing models | 170 |
 preritj/segmentation | Deep learning models for semantic segmentation of images | 100 |
 rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,821 |
 deng-cy/deep_learning_topology_opt | A toolkit for using machine learning to optimize complex geometries in simulations | 107 |
 simonkohl/probabilistic_unet | Reimplementation of a neural network model for conditional segmentation of ambiguous images | 546 |