> ## Documentation Index
> Fetch the complete documentation index at: https://training-docs.cerebras.ai/llms.txt
> Use this file to discover all available pages before exploring further.

# Overview

> The [cerebras.pytorch package](../cs-torch/cerebras-pytorch-api/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](../cs-torch/cerebras-pytorch-api/cerebras-pytorch-amp)         | Module for configuring and using automatic mixed precision |
| [cerebras.pytorch.optim](../cs-torch/cerebras-pytorch-api/cerebras-pytorch-optim)     | Module containing Cerebras compliant optimizers            |
| [cerebras.pytorch.sparse](../cs-torch/cerebras-pytorch-api/cerebras-pytorch-sparse)   | Module containing Cerebras sparsity routines               |
| [cerebras.pytorch.metrics](../cs-torch/cerebras-pytorch-api/cerebras-pytorch-metrics) | Module containing Cerebras compliant metrics               |