AWS SageMaker 推出训练会话 API 支持自定义训练循环

Amazon SageMaker AI 推出 Training Session APIs，旨在解决微调大语言模型时算法控制力与运维复杂度之间的权衡问题。该服务允许用户在本地机器编写自定义 Python 训练循环，通过 API 调用（如 forward_backward、optim_step）驱动托管的 GPU 会话执行计算，而无需自行管理集群或实例配置。API 分为 ServiceClient、TrainingClient 和 SamplingClient 三个层级，支持 GPT-OSS 20B 等基座模型及 LoRA 适配器加载。 计费模式从传统的按小时占用 GPU 转变为按 Token 处理量计费，仅在调用 GPU 绑定的操作时产生费用，数据加载、Reward 函数评分等逻辑在本地 CPU 运行不计费。文章展示了基于 GRPO 算法在 GSM8K 数学题上的最小化实现，以及 Code RL 食谱中针对 DeepCoder 数据集的竞赛编程任务案例，后者包含多轮工具调用和沙盒测试验证机制。 事实：SageMaker 提供了三种客户端对象来管理会话生命周期和具体操作，支持保存权重为 Model Package Group 中的版本。推断：这种架构适合需要非标准 Loss 函数、复杂 Reward 信号或动态 Curriculum 的研究团队，特别是那些希望快速迭代算法而不愿承担基础设施运维负担的 ML 工程师。猜测：对于依赖大量本地后处理（如复杂的 Agent 评估或人类反馈回路）的工作负载，这种按需付费模式可能比预留实例更具成本效益，但具体节省幅度取决于本地计算与云端推理的时间比例。