modelzoo.common.pytorch

dump_context	Provides DumpContext, a debug utility for dumping activations and gradients on a CPU/GPU run, and setting up debug names for dumped WSE activations to be automatically correlated.
half_dtype	Module which provides utilities for selecting 16-bit floating point representation.
input
input_utils
layers
model_utils
run_cstorch_eval_harness	Eval Harness run script built using the cstorch API
run_cstorch_flow	Generic run scripts build using the cstorch API
run_gpt_inference	GPT Inference script built using the cstorch API
run_utils	Utilities for running Cerebras Pytorch Models
utils	General purpose Pytorch Utilities

previous

modelzoo.common.model_utils.count_lines.count_lines

next

modelzoo.common.pytorch.dump_context