Getting Started
Concepts
Model Zoo
CS Torch
cerebras.pytorch
cerebras.pytorch.amp
cerebras.pytorch.optim
cerebras.pytorch.sparse
cerebras.pytorch.metrics
Cluster Monitoring
Fundamentals
Support
Bases: torch.nn.Module
torch.nn.Module
Methods
build_model
forward
previous
cerebras.modelzoo.fc_mnist.pytorch.model.MNIST
next
cerebras.modelzoo.fc_mnist.pytorch.prepare_data