| 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 |
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.
⌘I