在 TPU 上运行 Ray 第二部分：Ray AI 库

Ray 的高层库 Serve、Data 和 Train 可简化在 Google TPU 切片上运行 AI 工作负载。Ray Serve 通过简单的拓扑配置对大型多主机模型进行正确的成组调度，Ray Data 以原生 JAX batch 直接向加速器供数，消除数据加载瓶颈。JaxTrainer 则自动处理跨切片协调、checkpoint 与容错，简化 TPU 上的分布式训练。