谷歌与Meta合作 强化TPU对PyTorch的支持

谷歌正在推进一项新计划,使其人工智能芯片在运行 PyTorch方面表现更佳。谷歌目标是让其自研芯片张量处理单元(TPU)成为英伟达GPU的可运行替代方案,但仅有硬件并不足以推动广泛采用。知情人士称,该计划在谷歌公司内部被称为“TorchTPU”,旨在消除阻碍TPU芯片普及的一项关键障碍,即让已经基于 PyTorch软件构建技术基础设施的客户,能够在TPU上获得完全兼容、对开发者友好的体验。据悉,谷歌正在考虑将部分软件开源,以加快客户采用速度,谷歌为TorchTPU项目投入了更多组织资源、战略重视度。为加快开发进度,谷歌正与 PyTorch 的创建者和主要维护方Meta紧密合作。

—— 财联社
 
 
Back to Top