Python pruning

Open-source Python projects categorized as pruning

Top 15 Python pruning Projects

  1. Torch-Pruning

    [CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.

  2. InfluxDB

    InfluxDB – Built for High-Performance Time Series Workloads. InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.

    InfluxDB logo
  3. neural-compressor

    SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX Runtime

  4. aimet

    AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.

  5. mmrazor

    OpenMMLab Model Compression Toolbox and Benchmark.

  6. model-optimization

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

  7. nncf

    Neural Network Compression Framework for enhanced OpenVINOβ„’ inference

  8. Sparsebit

    A model compression and acceleration toolbox based on pytorch.

  9. SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

    SaaSHub logo
  10. sparsify

    ML model optimization product to accelerate inference.

  11. only_train_once_personal_footprint

    OTOv1-v3, NeurIPS, ICLR, TMLR, DNN Training, Compression, Structured Pruning, Erasing Operators, CNN, Diffusion, LLM

  12. wyng-backup

    Fast backups for logical volumes & disk images

  13. UPop

    [ICML 2023] UPop: Unified and Progressive Pruning for Compressing Vision-Language Transformers.

  14. delve

    PyTorch model training and layer saturation monitor (by delve-team)

  15. OWL

    Official Pytorch Implementation of "Outlier Weighed Layerwise Sparsity (OWL): A Missing Secret Sauce for Pruning LLMs to High Sparsity" (by luuyin)

  16. thesis

    Master's thesis, Uni Passau (by harshildarji)

  17. Pi-SqueezeDet

    Pruning SqueezeDet for inference on Raspberry PI

  18. SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

    SaaSHub logo
NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).

Python pruning discussion

Log in or Post with

Python pruning related posts

  • [P] Help: I want to compress EfficientnetV2 using pruning.

    1 project | /r/MachineLearning | 28 Jun 2023
  • [R] πŸ€–πŸŒŸ Unlock the Power of Personal AI: Introducing ChatLLaMA, Your Custom Personal Assistant! πŸš€πŸ’¬

    9 projects | /r/MachineLearning | 19 Mar 2023
  • Intel Textual Inversion Training on Hugging Face

    1 project | /r/StableDiffusion | 22 Dec 2022
  • [R] New sparsity research (oBERT) enabled 175X increase in CPU performance for MLPerf submission

    2 projects | /r/MachineLearning | 10 Sep 2022
  • [R] BERT-Large: Prune Once for DistilBERT Inference Performance

    2 projects | /r/MachineLearning | 16 Jul 2022
  • [R] How well do sparse ImageNet models transfer? Prune once and deploy anywhere for inference performance speedups! (arxiv link in comments)

    2 projects | /r/MachineLearning | 26 Jun 2022
  • Still worrying about model compression? MMRazor may work for you.

    1 project | /r/computervision | 30 Dec 2021
  • A note from our sponsor - InfluxDB
    www.influxdata.com | 21 Jun 2025
    InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now. Learn more β†’

Index

What are some of the best open-source pruning projects in Python? This list will help you:

# Project Stars
1 Torch-Pruning 3,038
2 neural-compressor 2,430
3 aimet 2,341
4 mmrazor 1,600
5 model-optimization 1,536
6 nncf 1,044
7 Sparsebit 331
8 sparsify 326
9 only_train_once_personal_footprint 304
10 wyng-backup 257
11 UPop 102
12 delve 81
13 OWL 67
14 thesis 17
15 Pi-SqueezeDet 2

Sponsored
InfluxDB – Built for High-Performance Time Series Workloads
InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.
www.influxdata.com

Did you know that Python is
the 2nd most popular programming language
based on number of references?