Hugging Face、Google Cloud TPUでPyTorchを直接実行できるTorchTPUを発表

新パッケージTorchTPUにより、torch_xlaやOptimum TPUを使わずにPyTorchコードをGoogle Cloud TPUで実行できる。

短信著者： OossaOossa公開日: 2026年10月20日 16:08 UTC最終更新: 2026年10月21日 1:35 UTC1 分で読める

Hugging Faceは、TPUをほかのデバイスと同じように扱えるネイティブPyTorchバックエンド「TorchTPU」を発表した。pipコマンド1つでインストールでき、device_map="tpu"を指定すれば、TPU上でTransformersやDiffusersのモデルを読み込める。このパッケージには、TPU向けに最適化したPallasカーネル、分散学習のサポート、さらなる高速化を実現するtorch.compileとの連携が含まれる。Hugging FaceのモデルをTPUで使う際に、Optimum TPUは不要になった。