PyTorch: Getting Started

PyTorch: Getting Started

If you are new to Cerebras, have access to Original Cerebras Installation and prefer to develop with PyTorch, begin with this quick-start guide before you get into in-depth development. It will introduce you to a typical user workflow on an Original Cerebras Installation.

The Original Cerebras Installation supports Pipelined execution only. You can run models that can fit entirely into the on-chip memory (below 1 billion parameters).

For PyTorch models, Cerebras uses PyTorch/XLA backend.

If you are ready to start developing / adapting your own PyTorch code for CS System

Skip to Workflow for PyTorch on CS for an in-depth development guide using PyTorch for Cerebras.