Cerebras PyTorch API#

The cerebras.pytorch package contains functions and data structures that can be used to configure and execute PyTorch models on a Cerebras Wafer Scale cluster.

cerebras.pytorch.amp

Module for configuring and using automatic mixed precision

cerebras.pytorch.optim

Module containing Cerebras compliant optimizers

cerebras.pytorch.sparse

Module containing Cerebras sparsity routines

cerebras.pytorch.metrics

Module containing Cerebras compliant metrics