Cerebras PyTorch API#
The cstorch-class-docs contains functions and data structures that can be used to configure and execute PyTorch models on a Cerebras Wafer Scale cluster.
cstorch-amp-class-docs
Module for configuring and using automatic mixed precision
cstorch-optim-class-docs
Module containing Cerebras compliant optimizers
cstorch-sparse-module-docs
Module containing Cerebras sparsity routines
cstorch-metrics-class-docs
Module containing Cerebras compliant metrics