跳到正文
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