Informes de IA · Amazon · 2026-10-22 · Blog / comentario
Escribe tu propio bucle de entrenamiento — sin gestionar GPUs — con las APIs de sesión de entrenamiento de Amazon SageMaker
Título original: Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
Las APIs de sesión de entrenamiento de Amazon SageMaker AI permiten a los desarrolladores escribir su propio bucle de entrenamiento de LLM (SFT, RL u objetivos personalizados) en un script de Python mientras el servicio ejecuta las GPUs. Operaciones como forward_backward(), optim_step() y sample() se convierten en llamadas a la API de SageMaker contra una sesión activa con estado, de modo que el modelo permanece cargado en las GPUs entre pasos. La publicación presenta el modelo de programación y tres clientes de SDK, y luego recorre un bucle GRPO sobre GSM8K y una carga de trabajo de aprendizaje por refuerzo sobre programación competitiva.