Module pruning 

Re-export aprender’s pruning primitives

Structs

MagnitudeImportance

Magnitude-based importance estimator.

MagnitudePruner

Simple magnitude-based pruner.

PruningResult

Result of a pruning operation with diagnostics.

SparseGPTImportance

SparseGPT importance estimator using Hessian-based saliency.

SparsityMask

Sparsity mask with validation.

WandaImportance

Wanda (Weights and Activations) importance estimator.

WandaPruner

Wanda-based pruner.

Enums

PruningError

Pruning operation errors with detailed context.

SparseTensor

Unified sparse tensor type.

SparsityPattern

Sparsity pattern constraints.

Traits

Importance

Core trait for importance estimation algorithms.

Pruner

High-level pruning interface.

Functions

generate_block_mask

Generate a block sparsity mask.

generate_column_mask

Generate a column sparsity mask.

generate_nm_mask

Generate an N:M structured sparsity mask.

generate_row_mask

Generate a row sparsity mask.

generate_unstructured_mask

Generate an unstructured sparsity mask based on importance scores.

sparsify

Apply a sparsity mask to a tensor and return sparse representation.