Industry
Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
Amazon SageMaker AI training session APIs let you write your own LLM training loop (SFT, RL, or custom objectives) in a Python script while the service runs the GPUs. This post introduces the programm
Amazon SageMaker AI training session APIs let you write your own LLM training loop (SFT, RL, or custom objectives) in a Python script while the service runs the GPUs. This post introduces the programming model and three SDK clients, then walks through a GRPO loop on GSM8K and a reinforcement learning workload on competitive programming.
Related
- Announcing new ways to customize models with SageMaker AI, without managing GPUs
- Multi-Region training with Amazon SageMaker HyperPod and Qumulo
- The generative AI customization spectrum: From prompt engineering to custom models on AWS
Source: AWS ML Blog | 2026-10-22