PyTorch

Habana SynapseAI is integrated with PyTorch and optimized for Habana Gaudi HPU. SynapseAI works under the hood, so as a developer you can still enjoy the same experience in PyTorch that you are accustomed to today.

• Getting Started with PyTorch and Gaudi
• PyTorch User Guide
• PyTorch Mixed Precision Training on Gaudi
• Distributed Training with PyTorch
• Habana Media Loader
• DeepSpeed Training with Gaudi
• Inference on Gaudi
• Model Performance Optimization
• Debugging and Troubleshooting Guide
• PyTorch Operators
• PyTorch CustomOp API