What problem does it solve?
Model pruning reduces LLM compute and memory usage while preserving accuracy by removing redundant parameters using pruning criteria such as Wanda and SparseGPT.
Core Features & Use Cases
- Untrained (one-shot) pruning: Compress models without retraining by using calibration activations or second-order information.
- Hardware-aware sparsity: Apply N:M sparsity (e.g., 2:4) to unlock accelerator-friendly speedups on supported GPUs.
- Structured vs unstructured options: Choose fine-grained sparsity for quality or structured sparsity for more reliable hardware execution.
Use cases: Deploying LLMs on constrained hardware, accelerating inference in production, reducing model footprint for edge/mobile deployment, and enabling faster serving without costly retraining.
Quick Start
Tell your AI agent to prune a Hugging Face Llama checkpoint to 50% sparsity using Wanda with a small calibration dataset, then save the resulting model for evaluation and deployment.