AWS Machine Learning Blog· Nicolas Jourdan·· 4 小时前AI 评分42
Amazon SageMaker AI 训练会话 API 发布：编写自定义训练循环而无需管理 GPU
Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs
AI 导读
Amazon SageMaker AI 推出训练会话 API，让开发者能在本地编写自定义训练循环，同时由服务管理 GPU 基础设施。该 API 提供 `forward_backward()`、`optim_step()` 和 `sample()` 等操作，支持监督微调、偏好优化和强化学习等自定义算法。
来源：AWS Machine Learning Blog · aws.amazon.com