VainF/Torch-Pruning
[CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.
This tool helps machine learning engineers and researchers make deep learning models more efficient by reducing their size and computational demands. You input an existing PyTorch deep neural network, and it outputs a smaller, pruned version of the same model. This is especially useful for deploying large models like those used in computer vision and natural language processing to devices with limited resources.
3,267 stars. Used by 1 other package. No commits in the last 6 months. Available on PyPI.
Use this if you need to optimize the size and speed of your large deep learning models, such as LLMs or vision transformers, for deployment or more efficient training.
Not ideal if you are working with very small, simple models where the overhead of pruning might outweigh the benefits, or if you only need to zeroize individual parameters without changing the model's overall structure.
Stars
3,267
Forks
374
Language
Python
License
MIT
Category
Last pushed
Sep 07, 2025
Commits (30d)
0
Dependencies
2
Reverse dependents
1
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/transformers/VainF/Torch-Pruning"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related models
peremartra/optipfair
Structured pruning and bias visualization for Large Language Models. Tools for LLM optimization...
horseee/LLM-Pruner
[NeurIPS 2023] LLM-Pruner: On the Structural Pruning of Large Language Models. Support...
CASIA-LMC-Lab/FLAP
[AAAI 2024] Fluctuation-based Adaptive Structured Pruning for Large Language Models
princeton-nlp/LLM-Shearing
[ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning
VITA-Group/LiGO
[ICLR 2023] "Learning to Grow Pretrained Models for Efficient Transformer Training" by Peihao...