Introduction

Intel® Extension for PyTorch* extends PyTorch* with the latest performance optimizations for Intel hardware. Optimizations take advantage of Intel XeMatrix Extensions (XMX) AI engines on Intel discrete GPUs. The extension provides easy GPU acceleration for Intel discrete GPUs through the PyTorch* xpu device.

For the detailed list of supported features and usage instructions, refer to Features.

Get Started

API Documentation

For detailed description of the Intel® Extension for PyTorch* APIs, refer to the API Documentation section.