跳到正文
AWS Machine Learning Blog· Nicolas Jourdan·· 1 小时前精选AI 评分63

Amazon SageMaker AI 推出 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,用户可在自己的 Python 循环中调用 forward_backward、optim_step 等原语完成 SFT 或 RL 训练,无需管理 GPU,按 token 付费。

推荐理由

用户可以在自己的 Python 循环中调用 sample、forward_backward 等原语完成自定义 RL 或 SFT,只需按 token 付费。

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