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

GitHub

2k stars
49 watching
148 forks
Language: Python
last commit: 6 months ago
Linked from 1 awesome list

automlcomputer-vision-algorithmsdeep-learning-algorithmsdeep-learning-librarydeep-learning-modelsimage-classificationkerasnlpobject-detectiononnxpruningpruning-algorithmspytorchsmaller-modelssparsificationsparsification-recipessparsitytensorflowtransfer-learning

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
intel/neural-compressor Tools and techniques for optimizing large language models on various frameworks and hardware platforms. 2,257
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,857
microsoft/archai Automates the search for optimal neural network configurations in deep learning applications 468
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. 248
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 Python package for gradient-based function optimization in machine learning 181
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 101
rentruewang/koila A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution 1,823
deng-cy/deep_learning_topology_opt Optimization code for topology optimization problems using machine learning and deep learning techniques 107
simonkohl/probabilistic_unet Reimplementation of a neural network model for conditional segmentation of ambiguous images 548