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 half dtype between float16 and bfloat16
input_utils
layers
model_utils
run_cstorch_flow	Generic run scripts build using the cstorch API
run_utils	Utilities for running Cerebras Pytorch Models
utils	General purpose Pytorch Utilities

previous

modelzoo.common.model_utils.sparsity

next

modelzoo.common.pytorch.dump_context