PyTorch
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