跳到正文
AWS Machine Learning Blog· Nicolas Jourdan·· 3 天前同新闻

AWS 推出 SageMaker 训练会话 API:自写训练循环,GPU 交由服务管理

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

AI 导读

AWS Machine Learning Blog 介绍 Amazon SageMaker AI 训练会话 API:用户可在 Python 脚本中编写自己的 LLM 训练循环,支持 SFT、RL 或自定义目标,由 SageMaker 运行 GPU。文章介绍编程模型和三个 SDK 客户端,并演示 GSM8K 上的 GRPO 循环及竞争性编程强化学习工作负载。

同一新闻,精选展示《AWS 为 SageMaker AI 无服务器模型定制新增训练会话并正式可用》

来源:AWS Machine Learning Blog · aws.amazon.com