PyTorch

User guides, API references and technical notes for PyTorch on the IPU and PyTorch Geometric on the IPU

User guides

PyTorch for the IPU: User Guide

User guide and API reference for PyTorch on the IPU

PyTorch Geometric for the IPU: User Guide

User guide and API reference for PyTorch Geometric on the IPU

PyTorch technical notes

Memory and Performance Optimisation on the IPU

Optimising high-performance machine learning models running on the IPU

Creating Custom Operations for the IPU

An overview of the steps for implementing a custom op in each of the frameworks available in the Poplar SDK

Optimising Temporary Memory Usage for Convolutions and Matmuls on the IPU

Using the “available memory proportion” option to optimise memory use or performance