Wrap your PyTorch code to run on Cerebras

This section describes how to port your PyTorch code to Cerebras.

• Port PyTorch to Cerebras
  • Option 1 (Easiest): Modify reference models in Cerebras Model Zoo git repository
  • Option 2 (Easy): Create new models leveraging Cerebras run function available in Cerebras Model Zoo
• PyTorch Runners
  • Using Custom Models with the PyTorch Runners
  • Creating Custom PyTorch Runners (Advanced)
  • Using Custom Runner
• Limitations of PyTorch on Cerebras
  • Floating Point Precision
  • Static Graphs
  • Learning Rate Scheduler
  • Eval metrics

previous

Limitations of the CerebrasEstimator

next

Port PyTorch to Cerebras