TPU v4芯片强如神话之刃钢铁般坚韧
Google I/O开发者大会在疫情的阴影下重燃希望,线上形式的回归标志着科技巨头对未来发展的坚定信心。CEO桑达尔·皮查伊在没有开发者见证的情况下宣布推出多项创新技术,其中包括全息视频聊天技术Project Starling和最新一代AI芯片TPU v4。
TPU v4:神话之刃,钢铁般坚韧
TPU v4是Google部署的最快系统,对Google来说是一个历史性的里程碑。与其前代相比,TPU v4性能提升2.7倍,每个Pod中有4096个单独芯片,可以达到1 exaFlOP级算力,这超过了全球任何超级计算机“富岳”的两倍性能。
MLPerf结果显示,256个TPU v4可以在1.82分钟内完成图像分类训练测试,与768张Nvidia A100卡、192块AMD Epyc 7742内核、512块华为AI优化的Ascend910芯片以及128块Intel Xeon Platinum 8168内核组合使用时间相当快。
除了用于训练模型外,TPU v4还将应用于MUM模型(Multitask Unified Model)和专为对话设计的LaMDA。这两种模型分别比BERT强1000倍和与人类不间断交流,是AI领域的一大突破。
自研五年更新四代
Google自2016年首次发布内部定制AI芯片以来,一直致力于推动这项技术。在过去几年的更新中,每一代产品都实现了显著提升,从最初仅适用于深度学习推理到能够进行机器学习模型训练和推理,再到现在这款能以惊人的速度处理数据的大型集群处理器。
随着每一次迭代更新,谷歌TPU逐渐成为行业中的领先者,不仅打破了GPU“垄断”地位,而且开辟了一条云端AI芯片新竞争格局。未来的世界或许充满变数,但Google TPU已经向我们揭示了一部分可能性的答案。