在 Intel Gaudi 2 上加速蛋白质语言模型 ProtST

Intel 与 MILA 借助 Optimum 库在 Intel Gaudi 2 上实现了蛋白质语言模型 ProtST 的推理与微调加速。在测试集运行 bfloat16 推理时，单张 Gaudi 2 取得与 NVIDIA A100 相同的 0.44 准确率，速度达 A100 的 1.76x。微调任务中单张 Gaudi 2 速度较 A100 提升 2.92x，且在 4 或 8 卡配置下呈近线性扩展。