Cerebras Model Zoo
Cerebras Model Zoo¶
This repository contains examples of common deep learning models that can be trained on Cerebras hardware. These models demonstrate the best practices for coding a model targeted at the Cerebras hardware so that you can take full advantage of this powerful compute engine.
The Model Zoo repository has two active branches. Reference the correct branch of the Model Zoo version that corresponds to your type of CS system installation.
If you are using the Cerebras Wafer-Scale Cluster, use the main branch of the Model Zoo repository. You can see a list of supported models here.
If you are using the Original Cerebras Installation, use the original_cerebras_installation branch of the Model Zoo repository. You can see a list of supported models here.