Google、Ray on TPU正式サポート発表

記事の3ポイント

1. 1

   TPU の公式サポート開始

   Ray 2.55 が Google Cloud TPU を正式にサポートし、開発者が Ray のタスクおよびアクター API をそのまま利用して分散 Python ワークロードを TPUs で実行できるようになった。

2. 2

   ハードウェア配置の自動化

   KubeRay Operator が GKE 上で多ホスト TPU スライスを維持するための厳格なネットワーク要件に対応し、基盤となるハードウェアレイアウトを自動的にプロビジョニングしてラベル付けする。

3. 3

   スライス配置プリミティブ

   Ray Core は slice_placement_group() プリミティブを通じてこれらのラベルを利用し、完全なスライスを原子的に予約することで、開発者がハードウェアトポロジー（例：4x4）を宣言するだけでジョブを展開できる。