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: about 2 years 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:

RepositoryDescriptionStars
intel/neural-compressorTools and techniques for optimizing large language models on various frameworks and hardware platforms.2,257
hiroyuki-kasai/sparsegdlibraryA collection of unconstrained optimization algorithms for sparse modeling in MATLAB53
minimaxir/automl-gsAutomates machine learning model creation and optimization for complex datasets1,857
microsoft/archaiAutomates the search for optimal neural network configurations in deep learning applications468
lge-arc-advancedai/auptimizerAutomates model building and deployment process by optimizing hyperparameters and compressing models for edge computing.200
google-research/sputnikA library of optimized GPU kernels for sparse matrix operations used in deep learning.248
xternalz/sdpointA deep learning method for optimizing convolutional neural networks by reducing computational cost while improving regularization and inference efficiency.18
dmlc/mxnet-memongerA tool for optimizing deep learning models to reduce memory usage without sacrificing performance308
brml/climinA Python package for gradient-based function optimization in machine learning181
herilalaina/mosaic_mlAutomated machine learning with tree search optimization16
balavenkatesh3322/nlp-pretrained-modelA collection of pre-trained natural language processing models170
preritj/segmentationDeep learning models for semantic segmentation of images101
rentruewang/koilaA lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution1,823
deng-cy/deep_learning_topology_optOptimization code for topology optimization problems using machine learning and deep learning techniques107
simonkohl/probabilistic_unetReimplementation of a neural network model for conditional segmentation of ambiguous images548