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
DownEncoderBlock2D
ResNetBlock2D
UNetMidBlock2D
UpDecoderBlock2D
VAEDecoder
VAEEncoder
VAEModel
utils
previous
modelzoo.vision.pytorch.dit.layers.schedulers.get_named_beta_schedule
next
modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D