Pallas 設計筆記

設計

• Pallas 設計
  • 簡介
  • Pallas：擴展 JAX 以用於核心
• Pallas 非同步操作
  • 背景 + 動機
  • 我們如何在 TPU 上實作分解的 Pallas 非同步操作？
  • 為什麼不這樣做？
  • 狀態的反撲
  • 結論