Yes. Cerebras's documentation describes the cstorch API as the way to "integrate Cerebras technology with PyTorch", and says you can bring your own PyTorch model and dataset.
What the documentation covers
The Cerebras training documentation offers:
- The cstorch API reference for connecting PyTorch code to Cerebras systems.
- A Model Zoo with pre-configured language, computer vision and multimodal models.
- Core workflow guides for data preprocessing, evaluation and training, and quickstarts for pre-training and fine-tuning.
- A setup guide for a Cerebras virtual environment.
The page does not list limitations, so confirm that your specific model and operators are supported before you plan around it.
Next steps
For the chip, see the Cerebras Wafer-Scale Engine, and for how it stacks up against NVIDIA, Cerebras vs NVIDIA. The wider field is in NVIDIA's main competitors in AI.
