MuKoe:開源的MuZero實作,分散式AI框架

Link:mukoe
MuKoe是一個完全開源的MuZero實作,使用Ray作為分散式編排器在GKE上執行。它提供了Atari遊戲的示例,並透過Google Next 2024的演講提供了程式碼庫的概覽。MuKoe支援在CPU和TPU上執行,具有特定的硬體要求,適合需要大規模分散式計算資源的AI研究和開發。

需求人群:

"適用於需要大規模分散式計算的AI研究和開發場景,如深度學習模型訓練、強化學習演算法實作等。"

使用場景示例:

使用MuKoe進行Atari遊戲的強化學習研究

在TPU上進行深度學習模型的訓練和推理

利用Ray框架進行大規模分散式計算任務

產品特色:

分散式計算

TPU加速學習

CPU推理

批量TPU推理

Ray叢集管理

返回頂端