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により、SFT、強化学習、カスタム目的を含むLLMのトレーニングループをPythonスクリプトで記述し、GPUの実行はサービス側が担える。forward_backward()、optim_step()、sample()といった操作がSageMaker API呼び出しとなり、対象はライブでステートフルなセッションであるため、モデルはステップをまたいでGPU上にロードされ続ける。本稿ではプログラミングモデルと3つのSDKクライアントを紹介したうえで、GSM8KでのGRPOループと競技プログラミングの強化学習ワークロードを解説する。