ray

Ray：分布式 AI 计算引擎，让大模型跑遍万台机器

想象这样一个场景： 你训练好了一个 GPT 级别的模型，想要在生产环境中同时服务 10 万用户——单机根本扛不住，分布式写代码又太复杂。Ray 就是来解决这个问题的：它把"如何把任务拆分到多台机器"这件事，变得像写普通 Python 函数一样简单。

背景：UC Berkeley 的十年磨一剑

Ray 诞生于加州大学伯克利分校（UC Berkeley）的 RISE 实验室，由该实验室的多位顶尖研究员于 2017 年正式对外发布。初衷很简单：当时没有现成工具能让研究者在自己的笔记本上写的 Python 代码，无缝扩展到拥有数千台机器的集群。研究者们要么手写分布式系统代码，要么使用 Hadoop/Spark 这类笨重的通用框架——两者都不适合现代 AI 工作负载。

随着大模型浪潮的到来，Ray 的价值被急剧放大。它的缔造者后来创办了 Anyscale 公司，提供基于 Ray 的商业化托管服务，2023 年已完成 C 轮融资，估值超 10 亿美元。Ray 本身则保持着强劲的社区增长，截至 2025 年初 GitHub 星标数已突破 42,000，成为 AI Infrastructure 领域最受关注的开源项目之一。

核心架构：一核 + 五库的分层设计

Ray 的架构分为两层：Ray Core 是底层分布式运行时，提供三个核心抽象；Ray AI Libraries 是上层一站式 AI 工具链，覆盖从数据处理到模型上线的全流程。

Ray Core 三大抽象：

• Tasks（任务）：将任意 Python 函数变成可并行执行的分布式任务，@ray.remote 装饰器一键标记，Ray 自动处理任务调度、失败重试和结果收集。
• Actors（角色） ：创建有状态的长驻进程，适合需要维护内部计数器、数据库连接池或模型实例的场景。相比任务，Actor 拥有固定地址，可反复调用。
• Objects（对象） ：Ray 集群中任意节点创建的数据对象，通过引用（Object Ref）在任意位置访问，底层自动处理跨节点传输和序列化。

Ray AI Libraries（五件套）：

代码架构：Python 为核心，Cython + C++ 加速

从代码仓库结构可以看出，Ray 是一个高度模块化的巨型项目：

• python/ray/ 包含所有核心模块，其中 air/（Ray AI Libraries Runtime）、rllib/（RLlib）、train/、tune/、serve/、data/ 分别对应五大 AI 库
• src/ 目录包含核心 C++ 扩展（_raylet.pyx），处理 gRPC 通信、对象存储和任务调度等高性能路径
• docker/ 目录提供完整的多架构镜像支持（CPU/GPU），build-docker.sh 脚本支持本地构建自定义镜像

整体代码质量较高：仓库有完善的 CI/CD（GitHub Actions）、完整的单元测试套件、详细的文档体系，Apache-2.0 许可证。

部署体验：pip 一行安装，Dashboard 可视化调试

Ray 的上手门槛在分布式框架中属于极低级别：

pip install ray              # 基础版
pip install "ray[default]"   # 含 Dashboard
pip install "ray[train]"     # 含分布式训练
pip install "ray[serve]"     # 含模型服务

安装后直接在 Python 脚本中使用 @ray.remote 装饰器即可实现分布式。Ray 内置 Ray Dashboard（http://localhost:8265），提供任务图谱可视化、资源使用率监控、日志聚合和 Actor 状态查看，对调试分布式应用非常有帮助。

硬件需求：轻量单机测试，重度依赖 GPU 集群

Ray 本身对硬件要求不高——Core 模块 2GB 内存即可运行一个单节点 Ray 集群。但 Ray 的真正价值在于横向扩展：

• CPU 训练/推理：Ray Serve 可部署 CPU 推理服务，8GB+ RAM 足够
• GPU 训练：Ray Train 默认需要 NVIDIA GPU，V100/A100/H100 均可，显存取决于模型规模
• Kubernetes：Ray 官方文档提供 KubeRay 项目，支持通过 Operator 在 K8s 上管理 Ray 集群，适合生产级大规模部署

行业意义：大模型 Infra 的基石组件

Ray 在 2024-2025 年的 AI 生态中扮演着关键角色：

• LLM 训练：Anthropic、OpenAI 等公司内部使用 Ray 管理大规模分布式训练
• LLM 服务：vLLM、Text Generation Inference（TGI）等推理框架底层依赖 Ray Serve
• AI Agent：多 Agent 协作框架（如 AutoGen、CrewAI）使用 Ray 作为任务编排层
• 数据处理：HuggingFace Datasets、Databricks Delta Lake 等项目集成 Ray Data

从 GitHub Topics 也能看出项目定位：large-language-models、llm-inference、distributed、hyperparameter-optimization 等标签精准覆盖了当前 AI 开发的核心需求。

局限与注意事项

Ray 并非银弹，以下场景需要谨慎评估：

1. 学习曲线：分布式概念（Tasks/Actor/Object）需要一定理解成本，不建议纯新手直接上生产
2. 资源调度：Ray 集群的资源分配需要手动配置，不当配置可能导致"节点空转"
3. 版本兼容性：Ray 与 PyTorch/TensorFlow 版本需匹配，建议使用官方 Docker 镜像避免依赖冲突
4. 监控复杂度：多节点 Ray 集群的监控比单进程复杂，需要配合 Prometheus + Grafana

总结

Ray 是当前 AI 基础设施领域最值得关注的开源项目之一。它用最少的代码改动，将单机 Python 程序扩展成分布式集群应用，同时提供从数据处理、训练、调优到上线的完整工具链。对于 AI 研究团队和 AI 应用公司来说，Ray 大幅降低了构建大规模 AI 系统的工程门槛，让团队可以把更多精力放在模型本身而非基础设施上。

图1：Ray 项目 GitHub 官方头像