← all repositories

VainF/Torch-Pruning

A PyTorch-based structural pruning framework that removes coupled parameters from deep neural networks including LLMs and vision foundation models.

Torch-Pruning
Velocity · 7d
+1.4
★ / day
Trend
steady
star history

Torch-Pruning implements the DepGraph algorithm to enable general-purpose structural pruning across diverse deep learning architectures. Unlike simple masking approaches, it physically removes parameters and their dependencies, reducing model size and computational cost. The toolkit supports off-the-shelf models from Huggingface, Timm, and Torchvision, targeting both large language models and vision models for efficient deployment.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.