PyTorch Model Porting

The purpose of this document is to guide Data Scientists to run PyTorch models on the Habana® Gaudi® infrastructure. It provides guidelines for modifying existing models to run on the platform and uses a basic example to show functionality.

• Porting PyTorch Models to Gaudi
• Placement of Ops on HPU
• Weight Sharing
• HPU Graphs for Training