跳到正文
PyTorch Blog(pytorch.org)· The TorchTPU Team†·· 11 天前精選品質評分88

Google 與 Meta 共同推出原生 PyTorch TPU 後端 TorchTPU

Introducing TorchTPU

重點導讀

Google 與 Meta 共同開源推出 TorchTPU,為 Google TPU 提供原生且符合 PyTorch 設計哲學的後端架構。

  • 核心特性:支援即時(eager)與編譯(compiled)模式,包含全新的 DeferAndFuse 執行模式,可顯著提升效能並減少高頻寬記憶體(HBM)佔用。
推薦理由

文章詳細說明 Google 與 Meta 合作推出原生的 PyTorch 執行後端,有助於開發者評估在 TPU 上無縫執行與遷移開源模型的效能表現。

信源出處pytorch.org

Introducing TorchTPU

本文為 888news 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。

開啟原文
來源: PyTorch Blog (pytorch.org)
開啟原文