AI 报告 · Amazon · 2026-10-22 · 博客 / 评论
编写自己的训练循环 — 无需管理 GPU — 借助 Amazon SageMaker 训练会话 API
原标题: Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
Amazon SageMaker AI 训练会话 API 让开发者能够在 Python 脚本中编写自己的 LLM 训练循环（SFT、RL 或自定义目标），同时由该服务运行 GPU。forward_backward()、optim_step() 和 sample() 等操作会变为针对实时有状态会话的 SageMaker API 调用，因此模型在各个步骤之间会一直保留在 GPU 上。这篇文章介绍了编程模型和三个 SDK 客户端，然后逐步演示了在 GSM8K 上运行的 GRPO 循环以及竞赛编程方面的强化学习工作负载。