跳到正文
原文
Hugging Face Blog·· 2023-01-02AI 评分36

用 Intel Sapphire Rapids 加速 PyTorch Transformers(上):分布式训练实战

Accelerating PyTorch Transformers with Intel Sapphire Rapids - part 1

AI 导读

Hugging Face 展示了如何在 AWS 的 Intel Sapphire Rapids(第四代 Xeon)裸金属实例上分布式训练 PyTorch Transformers,借助 Intel oneAPI CCL 和 IPEX 库,无需改动一行代码即可启用 AMX 指令加速矩阵乘法。

来源:Hugging Face Blog · huggingface.co