Port your own architecture#
- Port your PyTorch code using Cerebras Model Zoo
- (Early access) Port your code using Cerebras PyTorch API
- High Level Overview
- Importing
cstorch - Compiling a
torch.nn.Module - Initializing the Optimizer
- Initializing the DataLoader
- Using Gradient Scaling
- Constructing the training step
- Running Compile/Execution on a Cerebras Wafer Scale Cluster
- Step Closures
- Saving/Loading Checkpoints
- Training Loop
- Evaluation Metrics
- Full Example (Training)
- Full Example (Evaluation)
- Cerebras Model Zoo Supported Operations API