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

Amazon SageMaker AI 推出训练会话 API，自定义训练循环无需管理 GPU

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

AI 导读

Amazon SageMaker AI 推出训练会话 API，用户可在本地 Python 进程中驱动 GPU 会话，通过 forward_backward、optim_step、sample 等操作编写自定义训练循环，服务负责 GPU 分配、分布式执行、故障恢复等基础设施。

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