puzzletron

Modules

`modelopt.torch.puzzletron.activation_scoring`
`modelopt.torch.puzzletron.anymodel`	AnyModel: Architecture-agnostic model compression for HuggingFace models.
`modelopt.torch.puzzletron.block_config`
`modelopt.torch.puzzletron.build_library_and_stats`	Unified command that runs build_replacement_library followed by calc_subblock_stats.
`modelopt.torch.puzzletron.dataset`	Dataset preparation utilities for Puzzletron.
`modelopt.torch.puzzletron.entrypoint`	This module provides the main compression function for a model using MIP-based NAS search algorithm.
`modelopt.torch.puzzletron.mip`	MIP-based pruning: model optimization via mixed-integer programming.
`modelopt.torch.puzzletron.plugins`	Export utilities for Puzzletron models.
`modelopt.torch.puzzletron.pruning`	Structured pruning mixins and checkpoint utilities for Puzzletron.
`modelopt.torch.puzzletron.puzzletron_nas_plugin`	Puzzletron NAS plugin for the Modelopt framework (based on Puzzle algorithm: https://arxiv.org/abs/2411.19146).
`modelopt.torch.puzzletron.replacement_library`	Replacement library for Puzzletron layer substitution.
`modelopt.torch.puzzletron.scoring`	Validates and scores model compression solutions by evaluating puzzle solution candidates.
`modelopt.torch.puzzletron.sewing_kit`
`modelopt.torch.puzzletron.subblock_stats`	Subblock statistics collection for Puzzletron.
`modelopt.torch.puzzletron.tools`	Shared tools: logging, hydra config, checkpoint utilities, and validation helpers.
`modelopt.torch.puzzletron.utils`	Shared utilities for Puzzletron.