IT之家 10月21日の報道、Googleは現地時間火曜日、Google TPU向けのネイティブPyTorchフレームワークTorchTPUをGitHub上でオープンソース化したと発表した。これにより、GoogleはAmazonとともに、人工知能業界におけるNVIDIAの独占を打破しようとしている。現在、大多数のAIアプリケーションはデフォルトでNVIDIAチップ、すなわちグラフィックスプロセッサ(GPU)をベースに開発・稼働している。
PyTorchは人気のオープンソースツールライブラリで、AIアプリケーション開発者やスタートアップ企業がAIモデルの構築、訓練、デプロイに依存している。統計によれば、85%のAIアプリケーションがPyTorchを使用している。PyTorchは当初2016年にMetaによって開発されたが、当時は大規模言語モデルが世界を変えることなど誰もまったく予想していなかった。当時すでにNVIDIA GPUはAI研究の標準ハードウェアだったため、PyTorchの多くの機能はNVIDIA GPUへの対応専用に作られた。2022年、MetaはPyTorchをLinux財団に移管した。この非営利組織は多くの重要なオープンソースプロジェクトを管理している。
Googleが発表したTorchTPUにより、AIアプリケーションをGoogle自社開発のAIチップであるテンソル処理ユニット(TPU)上で稼働させることができる。また、既存のPyTorchアプリケーションをTPUへ、さらにはGoogleクラウドへ容易に移行できる。開発者は設定を少し変更するだけで、アプリケーションコードを書き直さずに移行を完了できる。
Amazonも現地時間月曜日に同様の発表を行い、自社のPyTorchバージョンTorchNeuronを公開した。TorchNeuronはコードを書き直すことなく、AIアプリケーションをAmazon自社開発チップのTrainiumおよびAmazonクラウド(AWS)に移植できる。Amazonはさらに、NVIDIAのPyTorchに対する支配力をさらに弱めると表明している。同社はPyTorchのメインプロジェクトにコードを提出しており、各種機能をNVIDIAチップから切り離し、PyTorchが標準で各種ハードウェアに対応できるようにしている。
ただしIT之家が注目したところによれば、Googleの一部の取り組みはAmazonを上回る規模かもしれない。Googleは、Meta、Hugging Face、Inferact(vLLMの開発元)、RadixArk(SGLangの開発元)などと直接協力し、TorchTPUの機能とコードベースがオリジナルのPyTorchと同様に、これらの主流AI開発ツールと良好に互換性を持つことを保証しているとしている。
さらにこの検索大手は、専用ツール「TPUエージェントキット」を発表し、開発者がアプリケーションをGoogleのチップとクラウドプラットフォームに移行し、移行中にエラーを調査・修正するのを支援している。Googleはまた、スマートフォン、ノートパソコン、組み込みデバイス上でAIモデルを稼働させるための広く使われているLiteRTツールをアップグレードし、PyTorchで開発されたAIアプリケーションをより良くサポートできるようにした。
以上の進展は、他のチップやクラウドプラットフォームに容易に移行できるというだけで、AIアプリケーション開発者が一斉にNVIDIAハードウェアを放棄することを意味するわけではない。しかし、AmazonとGoogleは明らかにNVIDIAの優位な地位を終わらせようと努力している。