重要

您正在查看 NeMo 2.0 文档。此版本对 API 和一个新的库 NeMo Run 进行了重大更改。我们目前正在将 NeMo 1.0 的所有功能移植到 2.0。有关先前版本或 2.0 中尚不可用的功能的文档,请参阅 NeMo 24.07 文档

库文档#

NeMo 是由 NVIDIA 开发的生成式 AI 框架,目标用户是使用 PyTorch 的研究人员和开发人员。其核心目的是提供一个强大且可扩展的框架,以促进生成式 AI 模型的设计和实现。NeMo 简化了对现有代码和预训练模型的访问,帮助来自工业界和学术界的用户加速其开发过程。开发者指南提供了关于 NeMo 的设计、实现和优化的广泛技术细节。

NeMo Aligner 是一个全面的工具包,专为高效的模型对齐而设计。它支持先进的对齐算法,包括 SteerLM、DPO 和来自人类反馈的强化学习 (RLHF),这些算法有助于调整语言模型,使其更安全、危害更小、更有益。该工具包适用于各种模型尺寸,并采用并行技术来优化模型对齐任务中的性能和资源效率。

NeMo Curator 是一个 Python 库,由多个可扩展的数据挖掘模块组成,专门用于整理自然语言处理 (NLP) 数据以训练大型语言模型 (LLM)。它使 NLP 研究人员能够有效地从庞大的、未经整理的网络语料库中提取高质量的文本,从而支持开发更准确和更强大的语言模型。