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

Lab · Write your own training loop, without managing GPUs, with Amazon SageMaker training session APIs

What is 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 programming model and three SDK clients, then walks through a GRPO l

The numbers

Open Write your own training loop, without managing GPUs, with Amazon SageMaker training session APIs →