intel-extension-for-pytorch
Intel® Extension for PyTorch* 提供优化功能,利用Intel® AVX-512 VNNI、AMX以及XMX AI引擎,提升Intel CPU和GPU上的深度学习性能。该扩展优化了大规模语言模型(LLMs),如LLAMA、GPT-J、GPT-NEOX等,支持多种量化方法(如FP32、BF16、INT8、INT4)。此外,自2.3.0版本起,还引入了模块级优化API,为定制模型优化提供了更多选项。