cerebras.modelzoo.common#
Various utility functions related to checkpoint saving and loading. |
|
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. |
|
Module which provides utilities for selecting 16-bit floating point representation. |
|
Helper functions related to configuration of the Optimizer. |
|
General purpose Pytorch Utilities. |
|
This is registry for the cerebras modelzoo |
|
Single flow BigCode Eval Harness run script. |
|
Generic run scripts build using the cstorch API. |
|
Eval Harness run script. |
|
GPT Inference script built using the cstorch API |
|
Utilities for running Cerebras Pytorch Models. |
|
Generic run scripts build using the cstorch API. |
|