
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.

Cerebras Model Zoo
This repository contains examples of common deep learning models demonstrating best practices for coding for the Cerebras hardware.
Developer Resources
Developer blogs
Cerebras Software Platform R1.7 is Out!
Our new release expands PyTorch support, releases code repositories for a range of extreme-scale GPT-style models, and introduces unique computer vision…
Chip,Machine Learning,Software,Cluster,SDK,Blog,HPC,Developer Blog
Harnessing the Power of Sparsity for Large GPT AI Models
Enabling innovation of novel sparse ML techniques to accelerate training and inference on large-scale language models.
Chip,Machine Learning,Software,Cluster,Blog,HPC,Developer Blog
Genomics in Unparalleled Resolution: Cerebras Wafer-Scale Cluster Trains Large Language Models on the Full COVID Genome Sequence
Our joint work with Argonne National Laboratory (ANL) and NVIDIA won the 2022 Gordon Bell Special Prize for HPC-Based COVID-19 Research.
Chip,Machine Learning,Software,Drug Discovery,Blog,HPC,Developer Blog
Real-Time Computational Physics with Wafer-Scale Processing
Cerebras and NETL achieve two orders of magnitude performance improvements for computational physics using a simple Python API for the Wafer-Scale Engine.
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!