Develop for the Cerebras System
With the Cerebras Software Platform, CSoft, you’ll spend more time pushing the frontiers of AI instead of optimizing distributed implementations.
Easily continuously pre-train multi-billion GPT-family models with up to an 20 billion parameters on a single device.
Get up and running quickly without changing your workflows. Utilize existing frameworks for your work or dive deep developing custom kernels – the Cerebras Software Platform is designed for flexibility and optimization.
PyTorch
Our PyTorch interface library is a simple wrapper for PyTorch program exposed through API calls that is easy to add as few extra lines of code for an existing Pytorch implementation.
TensorFlow
Integration of TensorFlow is via Cerebras Estimator, which is a wrapper class we developed based on standard TensorFlow Estimator and standard TensorFlow semantics.
SDK
The Cerebras SDK allows researchers to extend the platform and develop custom kernels – empowering them to push the limits of AI and HPC innovation.
Example Reference Implementations
This repository contains examples of common deep learning models demonstrating best practices for coding for the Cerebras hardware.
Developer Resources
Developer blogs
FAQ
Don’t see your question?Send us an email at developer@cerebras.net
To learn more about our programming model, please see our documentation – https://docs.cerebras.net/en/latest/cerebras-basics/how-cerebras-works.html#programming-model
- To get started with PyTorch please use this documentation
- To get started with TensorFlow please use this documentation
- Read the SDK technical overview white paper
- Please request access to our SDK here
Please find our example reference model implementation here: https://github.com/Cerebras/cerebras_reference_implementations. To get access to our full list, please contact us at developer@cerebras.net
Please sign up for our newsletter!