Agent SDK用来帮助用户快速训练AI智能体。
- 多种轨迹生成方法,支持插件。
- 融合调度高效利用显存。
如果第一次使用本软件,可以从快速入门中的样例开始上手,并确保按照其中步骤准备好相关环境和软件包。
如果对于相关流程已比较熟悉,可以直接跳转到Python接口说明获取需要的函数接口,加速数据处理流程。
Agent SDK软件架构如图1所示。
表 1 架构图模块介绍
| 模块 | 说明 |
|---|---|
| Agent SDK CLI接口 | 命令行接口。 |
| Agent SDK Python接口 | Python接口。 |
| Agent Engine注册 | 实现Agent的定义和Agent引擎的配置注册。 |
| 微调参数配置 | 微调阶段RL(Reinforcement learning)相关参数配置信息,比如train_iters等参数。 |
| Agentic RL | Agent RL微调的核心能力层,包括了Agent轨迹生成、多轮上下文记忆、分布式资源管理以及Agent训练推理运行时管理。 |
| GRPO-Trainer | 支持GRPO(Group Relative Policy Optimization)强化学习算法进行Agent RL微调。 |
| 产品系列 | 产品型号 | 操作系统版本 |
|---|---|---|
| Atlas A2 训练系列产品 | Atlas 800T A2 训练服务器 | Ubuntu 22.04 |
