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

Amazon SageMaker AI 推出训练会话 API，可自写训练循环而无需管理 GPU

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

AI 导读

Amazon SageMaker AI 推出 training session APIs，把 forward_backward()、optim_step()、sample() 等操作变成 API 调用，用户在本地 Python 中编写训练循环，GPU 分配、分布式执行、故障恢复和检查点由服务负责。

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