Google、PyTorch互換のTorchTPUを公開

TorchTPUは、大規模言語モデルが現在のAIブームを牽引する前にMetaが2016年に作ったPyTorchフレームワークの代替として登場した。PyTorchはNvidiaのGPUがすでにAI研究の主流ハードウェアだった時期に設計されたため、Metaが2022年にPyTorchをオープンソース化してLinux Foundationに管理を移した後も、ほとんどのAIアプリはこのハードウェアを対象とし続けてきた。

Googleの発表は、Amazon Web Servicesが今週初めにAmazonのTrainumチップ向けの独自PyTorch版TorchNeuronを発表したことに続くものだ。Amazonはまた、主要機能をNvidiaのGPUから切り離すコードを元のPyTorchプロジェクトに提供した。Googleはそうしなかったが、Meta、Hugging Face、Inferact Inc.、RadixArk Inc.と各社の開発ツールとの互換性で協力した点で、その面ではAWSより優位に立つ可能性がある。

TorchTPU以外にも、GoogleはAIアプリのGoogle Cloudへの移行を支援し、移行中のエラーを自動修正するTPUエージェントスイートを明らかにした。また、スマートフォンやノートPC、組み込み機器上で直接動くAIアプリを作るためのツールLiteRTを更新し、PyTorchベースのアプリと互換になった。同社は依然、数百万の開発者に現在のハードウェアから離れるよう説得する課題を抱えるが、今回の発表は、長期的にNvidiaの優位を少しずつ削れるという自信の表れだ。