Introduction
Get Started
Prepare your data
Cerebras PyTorch API
cerebras_pytorch
cerebras_pytorch.amp
cerebras_pytorch.optim
cerebras_pytorch.sparse
cerebras_pytorch.metrics
Cerebras Model Zoo
Cerebras Guides
Fundamentals
Support
checkpoint_converter
data
display_images
input
layers
model
modeling_dit
pipeline
run
sample_generator
Samples a large number of images from a pre-trained diffusion model using DDP.
sample_generator_dit
samplers
utils
previous
modelzoo.vision.pytorch
next
modelzoo.vision.pytorch.dit.checkpoint_converter