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.
Module for configuring and using automatic mixed precision
Module containing Cerebras compliant optimizers
Module containing Cerebras sparsity routines
Module containing Cerebras compliant metrics