utils
Modules
Dataset and dataloader utilities for Puzzletron. |
|
Small point-to-point object transport independent of a model runtime. |
|
Memory-efficient (vocab-chunked) cross-entropy + knowledge-distillation losses. |
|
Parsing and formatting utilities for configuration handling in model compression. |
|
Debug helper: hash the calibration samples each rank actually feeds to the model. |
|
Model validation and loss calculation utilities for single-GPU and multi-GPU setups. |
|
Puzzletron typed block configs -> vLLM/AnyModel config adapter. |
Shared utilities for Puzzletron.