最近，苹果公司宣布他们在开发人工智能模型时，选择了谷歌的张量处理单元（TPU），而不是行业内广泛使用的 NVIDIA 图形处理单元(GPU)。这一消息在苹果周一发布的技术论文中曝光，文中详细介绍了他们在 AI 能力方面的最新进展。
图源备注：图片由AI生成，图片授权服务商Midjourney
苹果透露，他们使用了谷歌云平台的 TPU 集群，具体是 v4和 v5p 版本，来训练他们的苹果基础模型（AFM）。在研究论文中，苹果提到，“AFM 模型是在 v4和 v5p 云 TPU 集群上进行预训练的，使用了 AXLearn 框架，这是一种基于 JAX 的深度学习库，专为公共云设计。”
苹果的这一决定对于即将推出的 “苹果智能” 计划尤为重要，该计划将驱动苹果生态系统内一系列 AI 驱动的新功能。值得注意的是，NVIDIA 在 AI 芯片市场中占据主导地位，其 GPU 是全球领先 AI 研究的首选。行业内普遍认为，NVIDIA 在 AI 芯片市场的份额高达95%。
苹果选择谷歌的 TPU 而非 NVIDIA 的 GPU，意味着大型科技公司正在探索其他选择。这也表明谷歌定制 TPU 的能力在不断提升，最初这些 TPU 是为内部使用而开发，现在则向外部客户提供。与此同时，Meta 近期也与 Groq 公司合作，利用其 LPUs 进行 Llama3.1的运行和推理。
在研究论文中，苹果指出，他们为服务器基础的 AI 模型使用了8192个 TPUv4芯片，而为 iPhone 及其他苹果设备设计的在设备 AI 模型则使用了2048个 TPUv5p 芯片。这些大规模的部署彰显了苹果在 AI 领域的雄心以及训练高级语言模型所需的计算能力。
虽然业界分析师推测苹果的选择可能与成本效率、性能优势或战略合作有关，但苹果并未具体说明为何选择谷歌的 TPU。随着苹果开始向测试用户推出 “苹果智能” 框架的部分功能，显然他们打算在 AI 领域更积极地竞争。
划重点:
1. 📊 苹果选择谷歌 TPU 训练 AI 模型，而非 NVIDIA GPU，标志着 AI 硬件选择的多样化。
2. 🌐 苹果的 “苹果智能” 计划将利用 TPU 进行 AI 功能开发，彰显公司在 AI 领域的雄心。
3. 🤖 苹果部署了大量 TPU 芯片以满足其 AI 模型的训练需求，显示出其在技术上的投入与决心。