メインコンテンツへスキップ
見出し画像

Google、TPU向け「TorchTPU」をオープンソース化:PyTorch対応でNVIDIAのCUDA優位に挑む

Googleは、2026年10月20日、TPUをPyTorchのネイティブデバイスとして扱う「TorchTPU」をオープンソース化しました。モデルやテンソルの配置先をGPUのcudaからtpuへ変え、Eager Mode、torch.compile、分散学習といったPyTorchの標準的な操作でTPUを利用するための基盤です。Google独自の開発環境へ利用者を移すのではなく、TPUの方をPyTorchへ寄せたことが、今回の発表の中心になります。〔資料1・2〕

あなたへのおすすめ