puzzletron

Modules

modelopt.torch.puzzletron.anymodel

AnyModel: Architecture-agnostic model compression for HuggingFace models.

modelopt.torch.puzzletron.build_library_and_stats

Unified command that runs build_replacement_library followed by calc_subblock_stats.

modelopt.torch.puzzletron.dataset

modelopt.torch.puzzletron.puzzletron

This module provides the main compression function for a model using MIP-based NAS search algorithm.

modelopt.torch.puzzletron.sewing_kit

modelopt.torch.puzzletron.tools