AI@NEWS

Ray 2.55支持TPU调度

Ray 2.55新增对Google Cloud TPU的一等支持,可用KubeRay管理多主机TPU切片。

推荐理由:实质变化是Ray把TPU切片的原子预留抽象进slice_placement_group,降低多主机ICI拓扑调度复杂度,使JAX/训练任务更易迁移到Ray生态。

阅读原文