PyTorchモデルをTPUで動かすTorchTPUのリポジトリがGitHubで公開(github.com)

- このリポジトリは、PyTorchで作られたAIモデルをGoogleのTPU（AI演算に特化したチップ）で実行できるようにする接続プログラムである。READMEには、演算の実装コード、実行結果を再利用するコンパイルキャッシュ、分散学習機能が含まれると書かれている。 - READMEは、データ並列とテンソル並列の分散学習、torch.compile()との連携、Llama、Qwen、ResNetなどのモデル例を機能として挙げている。原文には速度や精度を測定した結果がないため、性能は確認されていない。 - インストールは、pipでバージョン0.2.0を入手する方法と、Bazelビルドツールで直接ビルドする方法が案内されている。対応するPythonのバージョンは3.11以上と書かれている。