Google、TPU向けネイティブPyTorchバックエンド「TorchTPU」をオープンソース化
原題: Google open-sources TorchTPU, a native PyTorch back end for TPUs
AI要約
生成・訂正方針GoogleがTPU(Tensor Processing Unit)向けにネイティブなPyTorchバックエンドであるTorchTPUをオープンソースとして公開した
重要ポイント
- •TPU環境でPyTorchモデルをネイティブに実行するためのバックエンドとして開発された
- •GitHub上のgoogle-pytorch/torch_tpuリポジトリを通じて開発コミュニティに貢献する仕組みが提供されている