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.

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