AWS Machine Learning Blog· Nicolas Jourdan·2026-10-23 02:00· 3 小时前AI 评分18

用 Amazon SageMaker 训练会话 API 编写自己的训练循环，无需管理 GPU

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

AI 导读

Amazon SageMaker AI 的训练会话 API 让开发者用 forward_backward()、optim_step()、sample() 等 API 调用自行编排 SFT、偏好优化或 RL 训练循环，模型在 GPU 上保持加载，服务负责分布式执行、梯度同步与故障恢复。

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