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
After mapping the TensorSpecs to tensors/callables, return them directly as leaf nodes instead of reconstructing a dictionary when unflattening.
previous
cerebras.modelzoo.data.common.SyntheticDataProcessor.custom_dict_flatten
next
cerebras.modelzoo.data.common.SyntheticDataProcessor.SyntheticDataProcessor