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

previous

Glossary

next

Prepare your data