Overview

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