Anyscale 和 Google Cloud 已合作推出对 Google TPU 上的 PyTorch 的原生支持,利用 Ray 分布式计算框架。此集成允许机器学习团队通过单一 Kubernetes 控制平面管理跨 GPU 和 TPU 的分布式 PyTorch 训练和推理工作负载。新方法通过启用每 TPU 芯片一个工作程序的模型,简化了从 GPU 到 TPU 的迁移,该模型模仿了熟悉的 GPU 设置,并允许进行直接的代码调整。 AI
影响 简化了机器学习团队在 Google Cloud 的 TPU 基础设施上进行分布式 AI 训练和推理。
排序理由 这是关于特定硬件上的分布式计算框架的产品集成公告,而不是核心 AI 模型发布或研究论文。
- Anyscale, Inc.
- Gemini
- Google Cloud
- Google Kubernetes Engine
- Google TPUs
- Jax
- KubeRay
- Kubernetes
- PyTorch
- Ray
- TorchTPU
- vLLM
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →