AWS Machine Learning Blog· Nicolas Jourdan·2026-10-23 02:00· 4 小时前精选AI 评分78

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

Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs

AI 导读

Amazon SageMaker 推出训练会话 API，允许用户在本地编写自定义训练循环，由服务托管 GPU 执行采样、前向传播和梯度更新。该功能支持按 Token 计费，提供 SFT 和 RL 等预置食谱，并附带 GSM8K 数学推理与 Code RL 编程竞赛的实战示例。

推荐理由

通过训练会话 API 将 GPU 管理抽象化，允许用户编写自定义训练循环，实现按 Token 计费并降低复杂算法的运维门槛。

来源：AWS Machine Learning Blog · aws.amazon.com