Skip to main content

oxicuda_quant/pruning/
mod.rs

1//! # Pruning
2//!
3//! Weight-pruning strategies for model compression.
4//!
5//! | Module      | Strategy                                               |
6//! |-------------|--------------------------------------------------------|
7//! | `mask`      | [`SparseMask`] — boolean weight mask primitives        |
8//! | `magnitude` | [`MagnitudePruner`] — unstructured L1/L2 pruning       |
9//! | `structured`| [`StructuredPruner`] — channel / filter / head pruning |
10
11pub mod magnitude;
12pub mod mask;
13pub mod structured;
14
15pub use magnitude::{MagnitudeNorm, MagnitudePruner};
16pub use mask::SparseMask;
17pub use structured::{PruneGranularity, StructuredPruner};