AI 训练格局转变:苹果弃用 NVIDIA,转投 Google TPU
科技巨头苹果公司最近发布的技术论文显示,其人工智能系统 Apple Intelligence 的模型预训练并非依赖于 NVIDIA 芯片,而是采用了 Google 云端设计的芯片。
NVIDIA 的图形处理单元(GPU)以其高昂的价格和旺盛的需求而闻名,一直是众多科技公司进行 AI 训练的首选。然而,苹果的此次选择表明,大型科技公司在 AI 基础设施方面正转向多元化。
苹果在论文中详细阐述了使用 TPU v4 和 TPU v5p 芯片进行模型预训练的过程。据悉,苹果的设备端 AI 模型是在 2048 个 TPU v5p 芯片上训练的,而服务器端模型则使用了 8192 个 TPU v4 芯片。
值得注意的是,尽管苹果最初采用了 Google 的硬件,但该公司已部署基于其自研 Apple Silicon 芯片的硬件来处理 Apple Intelligence 的查询。此外,苹果还启动了“Project ACDC”项目,旨在优化数据中心内的 AI 应用。
苹果计划在未来两年内斥资 50 亿美元用于 AI 服务器的升级,并购买数万台 AI 服务器。同时,该公司还收购了一家专注于压缩 AI 查询数据的公司,以提升其在 AI 领域的竞争力。