Ray 是一个开源框架,用于管理、执行和优化计算需求,由 Anyscale 提供支持。它被定位为“AI 计算引擎”,目标是为任意规模、任意加速器上的分布式工作负载精确编排基础设施,从而帮助团队开发 AI 时获得规模、性能和效率。
用途与定位 Ray 面向 AI、机器学习和生成式 AI 工作负载。官方描述指出,AI 日益复杂,数据模态和模型、框架、加速器层出不穷,缺乏有效基础设施会导致上线慢、资源利用率低和成本激增。Ray 作为 AI 计算引擎,旨在支持任意 AI 工作负载,应对“AI 复杂性之墙”。
主要能力
- Python 原生与并行 Python 代码:Ray 是 Python 原生的,可将任意 Python 代码扩展和分布式化,适用于仿真、回测等场景。
- 多模态数据处理:可处理结构化与非结构化数据,包括图像、视频、音频等。
- 分布式训练:用一行代码即可分布式运行生成式 AI 基础模型、时间序列模型以及 XGBoost 等传统 AI/ML 模型,并兼容开发者选择的框架。
- 模型部署:Ray Serve 用于部署模型和业务逻辑而非实例,支持独立扩缩容和分数资源,支持从 LLM 到稳定扩散模型、目标检测模型等各类 ML 模型。
- 批量推理:利用异构计算简化离线批量推理流程,可在同一流水线中使用 CPU 和 GPU,提高利用率并降低成本。
- 强化学习:Ray RLlib 支持生产级、高度分布式的强化学习工作负载,同时保持统一简单的 API。
- 生成式 AI 工作流:支持多模态模型、RAG 应用等端到端 GenAI 工作流。
- LLM 推理与微调:支持大语言模型在线和批量推理,可无缝扩展;并支持大规模微调大语言模型(源自 ChatGPT 背后的框架)。
关键概念与生态 Ray Core 提供少量核心原语(任务、Actor、对象),用于构建和扩展分布式 Python 应用。Ray 还提供面向开发者的高层 ML 库,可无缝扩展数据处理、训练和 serving 等任意 AI 工作负载。完整 ML 生态支持在 Ray 上运行端到端 AI/ML 工作流,提供部署 Ray 集群、调试优化应用以及与常见工具和框架集成的易用工具。
Anyscale 与使用方式 Anyscale 是 Ray 的全托管 AI 平台,由 Ray 创始团队打造,提供企业治理、高级开发者工具和专家支持。官网提供免费试用入口,并建议在 Anyscale 上运行 Ray 以获得全托管一体化体验。用户也可加入 Ray Slack 社区交流。GitHub 仓库提供下载。
适用场景 官方列出的场景包括:大规模模型训练、多模态数据处理、批量推理、强化学习、LLM 在线与批量推理、LLM 微调、RAG 应用等。行业领先者使用 Ray 加速 AI 运营,官网提及部分成果数据(如更多训练数据、CPU 核心用于在线模型服务、基础模型训练参数量、数据处理成本降低、从 Spark 切换到 Ray 用于 GPU 批量推理的成本下降、GPU 利用率提升与成本降低、每月训练模型数量、更快迭代交付 100+ 生产模型等),但未提供具体客户名称与详细条件。
限制与缺失信息 官方资料未提供具体收费价格、免费额度细节、平台支持的操作系统或部署环境限制、具体性能基准的完整测试条件。部分功能(如具体框架兼容列表、集群规模上限)暂未提供。