Ways to port your model#

Cerebras Model Zoo provides tools to facilitate porting your model to Cerebras ecosystem, from using one of Cerebras Model Zoo models out of the box, to writing your own implementation using Cerebras APIs. This section provides documentation for each of these cases:

Stage

Prepare your data

Define model architecture

Beginner: Getting started with Cerebras Ecosystem

Use dataloaders and preprocessing scripts in Cerebras Model Zoo (PT, TF)

Use model implementation in Cerebras Model Zoo and customize hyperparameters in the params yaml file

Intermediate: Use your own data and hyperparameters

Customize preprocessing scripts and dataloaders (PT, TF)

Advanced: Define your own model using Cerebras Model Zoo tools

Port your PyTorch or TensorFlow code using run function in Cerebras Model Zoo and Cerebras Model Zoo Supported Operations API

Experimental: Define your model using Cerebras PyTorch API

Have more flexibility porting your code with Cerebras PyTorch API