Отчёты по ИИ · Amazon · 2026-10-22 · Блог / комментарии
Пишите собственный цикл обучения — без управления GPU — с API обучающих сессий Amazon SageMaker
Оригинальное название: Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
API-интерфейсы тренировочных сессий Amazon SageMaker AI позволяют разработчикам писать собственный цикл обучения LLM (SFT, RL или с собственными целями) в Python-скрипте, пока сервис управляет GPU. Операции вроде forward_backward(), optim_step() и sample() превращаются в вызовы SageMaker API к активной сессии с сохранением состояния, поэтому модель остаётся загруженной на GPU между шагами. В посте представлена модель программирования и три SDK-клиента, а затем разбираются цикл GRPO на GSM8K и рабочая нагрузка по обучению с подкреплением на спортивном программировании.