AWS Machine Learning Blog· Nicolas Jourdan·· 2 小时前
Amazon SageMaker 推出 training session APIs:自写训练循环而无需管理 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 调用,让用户自己编写 SFT、偏好优化或 RL 训练循环,同时由服务托管 GPU、分布式执行、故障恢复等基础设施。
来源:AWS Machine Learning Blog · aws.amazon.com