Обновление Amazon SageMaker AI: Script mode в SDK v3 для моделей и развертывания

В 2021 году был представлен режим скриптов в Amazon SageMaker, позволяющий запускать собственный код обучения и инференса без необходимости сборки Docker-образов. В новой версии SDK v3 произошла кардинальная переработка: теперь используются единые классы ModelTrainer и ModelBuilder, что упрощает рабочие процессы. Основная новация — механизм SourceCode, который синхронизирует локальный код в контейнере во время запуска, что позволяет быстро вносить изменения и повторно запускать обучение без пересборки образов.

В статье подробно описываются два примера: обучение и развертывание модели случайного леса на scikit-learn, а также дообучение модели Stable Diffusion 3.5 с помощью LoRA на нескольких GPU. В обоих случаях используется единый API, что облегчает работу с различными фреймворками и задачами.

Ключевые преимущества нового SDK — ускорение итераций, полный контроль над контейнером, универсальный интерфейс для разных фреймворков и возможность работы с любыми образами контейнеров, включая собственные. Также реализована поддержка работы с секретами через AWS Secrets Manager и возможность использования YAML-рецептов для настройки гиперпараметров.

Обновленный SDK подходит для задач от классического машинного обучения до генеративных моделей и распределенного обучения, что делает его универсальным инструментом для разработчиков и исследователей AI.