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.