Intel® Extension for PyTorch* extends PyTorch* with up-to-date features optimizations for an extra performance boost on Intel hardware. Optimizations take advantage of Intel® Advanced Vector Extensions 512 (Intel® AVX-512) Vector Neural Network Instructions (VNNI) and Intel® Advanced Matrix Extensions (Intel® AMX) on Intel CPUs as well as Intel Xe Matrix Extensions (XMX) AI engines on Intel discrete GPUs. Moreover, Intel® Extension for PyTorch* provides easy GPU acceleration for Intel discrete GPUs through the PyTorch* xpu device.
Features
- Large Language Models (LLMs) Optimization
- Documentation available
- Examples available
- Python package for extending the official PyTorch
- Can easily obtain performance on Intel platform
- Licensed under the Apache License
Categories
Machine LearningLicense
Apache License V2.0Follow Intel Extension for PyTorch
Other Useful Business Software
Host LLMs in Production With On-Demand GPUs
Deploy your model, get an endpoint, pay only for compute time. No GPU provisioning or infrastructure management required.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of Intel Extension for PyTorch!