utils

Modules

modelopt.torch.puzzletron.utils.data

Dataset and dataloader utilities for Puzzletron.

modelopt.torch.puzzletron.utils.distributed_object

Small point-to-point object transport independent of a model runtime.

modelopt.torch.puzzletron.utils.dummy_modules

modelopt.torch.puzzletron.utils.flash_kd

Memory-efficient (vocab-chunked) cross-entropy + knowledge-distillation losses.

modelopt.torch.puzzletron.utils.misc

modelopt.torch.puzzletron.utils.parsing

Parsing and formatting utilities for configuration handling in model compression.

modelopt.torch.puzzletron.utils.sample_hash

Debug helper: hash the calibration samples each rank actually feeds to the model.

modelopt.torch.puzzletron.utils.validation

Model validation and loss calculation utilities for single-GPU and multi-GPU setups.

modelopt.torch.puzzletron.utils.vllm_adapter

Puzzletron typed block configs -> vLLM/AnyModel config adapter.

Shared utilities for Puzzletron.