谷歌新一代云AI芯片TPU v5p发布,性能提升2 8倍

小夏 汽车 更新 2024-01-28

12月7日凌晨,谷歌(Google)在发布多模态大模型Gemini的同时,还推出了全新的面向云端的AI加速TPU V5P,这也是谷歌迄今为止功能最强大、性价比最高的TPU(Cloud Tensor Processing Unit)。

据报道,每个 TPU V5P Pod 由多达 8,960 个芯片组成,使用最高带宽的芯片到芯片连接(每个芯片 4,800 Gbps)互连,以确保快速传输速度和最佳性能。

在 AI 性能方面,TPU V5P 能够提供 459 teraflops(每秒 459 万亿次浮点运算)的 Bfloat16(16 位浮点格式)性能或 918 teraops(每秒 918 万亿次整数运算)的 INT8(8 位整数执行)性能,支持 95GB 高带宽内存,并且能够在 276 TBS 用于数据传输。

与 TPU V4 相比,新发布的 TPU V5P 具有两倍的 FLOPS(每秒浮点运算)和三倍的高内存带宽提升,这在人工智能领域是惊人的。

此外,在模型训练方面,TPU v5P显示2.在LLM(大型语言模型)训练速度上与 TPU V5E 相比,代际改进了 8 倍,提高了 50%。 谷歌也在榨取更多的计算能力,因为 TPU V5P “就每个 pod 的总可用 FLOP 而言,可扩展性是 TPU V4 的 4 倍”。

综上所述,TPU v5P 与 TPU v4:

浮点运算增加 2 倍(459 TFLOPS BF16 918 TOPS INT8)。

内存容量比 TPU v4 (95 GB HBM) 高 3 倍。

LLM训练速度提升28次。

嵌入密集模型训练为 19次。

带宽增加 225 倍(2765 GB 秒对 1228 GB 秒)。

带宽是芯片到芯片互连的两倍(4800 Gbps vs. 2400 Gbps)。

谷歌在拥有最好的硬件和软件资源方面取得了明显的成功,这就是为什么该公司拥有一台人工智能超级计算机,这是一组旨在协同工作以实现现代人工智能工作负载的元素。 谷歌集成了性能优化计算、优化存储、液冷等功能,充分利用了庞大的功能,输出性能确实处于行业领先水平。

在软件方面,谷歌已经加强了对开放软件的使用,以调整其人工智能工作负载,以确保其硬件的最佳性能。

以下是 AI Hypercomputer 新软件资源的概要:

对 Jax、TensorFlow 和 PyTorch 等流行 ML 框架的广泛支持开箱即用。 JAX 和 PyTorch 都由 OpenXLA 编译器提供支持,用于构建复杂的 LLM。 XLA 作为基础骨干,支持创建复杂的多层模型(使用 PyTorch XLA 在云 TPU 上进行 LLAMA 2 训练和推理)。 它优化了跨各种硬件平台的分布式架构,确保针对不同的 AI 用例(AssemblyAI 利用 JAX XLA 和 Cloud TPU 进行大规模 AI 语音)的易用且高效的模型开发。

开放独特的多切片训练和多主机推理软件,分别使扩展、训练和服务工作负载变得顺畅和简单。 开发人员可以扩展到数以万计的芯片,以支持要求苛刻的 AI 工作负载。

与 Google Kubernetes Engine (GKE) 和 Google Compute Engine 的深度集成提供了高效的资源管理、一致的操作环境、自动扩展、节点池自动配置、自动检查点、自动恢复和及时的故障保护。

谷歌对人工智能的革命性方法通过其新的硬件和软件元素显而易见,这些元素将打破限制行业的障碍。 看看新的Cloud TPU V5P处理单元将如何帮助AI超级计算机正在进行的AI开发,这将是一件有趣的事情,但有一点是肯定的,它们肯定会加剧竞争。

编辑:新知勋-流氓剑。

相似文章

    光子芯片是未来新一代信息产业的基础设施和核心支撑

    光子芯片是未来新一代信息产业的基础设施和核心支撑。月日,来自意大利米兰理工大学 比萨圣安娜大学 英国格拉斯哥大学和美国斯坦福大学的科学家联合研发了一种新的光子芯片。光子芯片时代即将到来,光子芯片是一种以光为载体进行信息传输的芯片,与传统电子芯片相比具有诸多优势,光子芯片技术的发展也为中国芯片提供了机...

    新一代工业节能空调

    工业节能空调不仅是一种设备,更是科技创新的杰出代表。通过引入智能管理和高效制冷技术,正在改变传统工厂车间的温控模式,成为提高生产效率 实现节能降耗的有力助手。工业节能空调台智能管理,精准管控 工业节能空调采用先进的智能管理系统,通过传感器实时监控车间内的温度 湿度等环境参数。根据实际需要对系统进行精...

    全新一代东风本田CR Ve PHEV搭载e HEV,第4代i

    近日,本田汽车推出了新一代东风本田CR VE PHEV和E HEV,搭载 i MMD混合动力技术。该技术是本田自主研发的电动混合动力系统,经过多年的创新和积累,已成为国内自主品牌中最先进的混合动力技术之一。第一代本田IMA混合动力系统在世界范围内取得了巨大的成功,成为当时世界上耗油量最低的量产车。如...

    采用快干技术的新一代纸巾

    洗手已经是习日常生活中的卫生习,但大多数人通常洗手后用毛巾 干手器或纸巾擦干双手。但实际上,纸纸是最合适的擦手方式,但普通纸纸吸水少,需要几张纸才能完全擦干手,还可能破碎产生杂物,粘在手上,也正是因为普通纸纸质量差,人们还没有养成用纸纸擦手的好习。QuickDry强大的速干技术的出现很好地解决了这个...

    TCL空调荣获“南山奖” 引领新一代空调创新转型

    月日,空气净化行业最高荣誉 南山奖 榜单在广州揭晓,TCL空调凭借新风空调科技创新荣获 南山奖 南山奖 以共和国勋章获得者 中国工程院院士钟南山的名字命名,堪称空气净化领域的 奥斯卡 该奖项由中国工程院院士位院士和多位专家组成的品牌评价专业技术委员会联合评选,采用网络投票方式,旨在表彰综合表现优异的...