​
Model Zoo API

The Model Zoo API provides a robust set of reference layers for both transformers and vision models within the cerebras.modelzoo.layers module. This module serves as a foundational resource for building and customizing your own models.

For task-specific requirements, the Model Zoo API offers specialized modules. The cerebras.modelzoo.models.nlp module includes layers and components tailored for natural language processing tasks, while the cerebras.modelzoo.models.vision module caters to computer vision applications. These modules streamline the development process by offering pre-configured, high-performance layers designed to meet the unique demands of NLP and vision tasks.

​
Trainer API

The Trainer API provides a simple but flexible interface to run any type of training and/or validation. The Trainer is designed to be easily extendable through the Callback mechanism and most of the Trainer’s core functionality is already implemented via a set of core callbacks available to users. Once configured, a training or validation loop can be run by simply calling trainer.fit or trainer.validate.