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 迴圈，以及競賽程式設計方面的強化學習工作負載。