Link:mukoe
MuKoe是一個完全開源的MuZero實作,使用Ray作為分散式編排器在GKE上執行。它提供了Atari遊戲的示例,並透過Google Next 2024的演講提供了程式碼庫的概覽。MuKoe支援在CPU和TPU上執行,具有特定的硬體要求,適合需要大規模分散式計算資源的AI研究和開發。
需求人群:
"適用於需要大規模分散式計算的AI研究和開發場景,如深度學習模型訓練、強化學習演算法實作等。"
使用場景示例:
使用MuKoe進行Atari遊戲的強化學習研究
在TPU上進行深度學習模型的訓練和推理
利用Ray框架進行大規模分散式計算任務
產品特色:
分散式計算
TPU加速學習
CPU推理
批量TPU推理
Ray叢集管理