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
This is registry for the cerebras modelzoo
Classes
Registry
previous
cerebras.modelzoo.common.pytorch_utils.SampleGenerator
next
cerebras.modelzoo.common.registry.Registry