AWS Machine Learning Blog· Nicolas Jourdan·· 3 小时前
Amazon SageMaker AI 推出 training session API，可在托管 GPU 上自写训练循环
Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
AI 导读
Amazon SageMaker AI 的 Serverless 模型定制新增 training session API，把抽象层从算法下移到算力，开发者用 forward_backward、optim_step、sample 等调用在本地 Python 进程中编写 SFT、偏好优化或 RL 循环，GPU 分配、分布式执行与容错由服务承担。
来源：AWS Machine Learning Blog · aws.amazon.com