性能强化学习研究工程师
岗位摘要
提升模型为加速器安全编写正确且快速代码的能力；发明、设计并实现强化学习环境与评估；开展实验并参与制定研究路线图；将研究成果交付并应用于训练运行；与 Anthropic 内外部的其他研究人员、工程师及性能工程专家协作
岗位职责
- 提升模型为加速器安全编写正确且快速代码的能力
- 发明、设计并实现强化学习环境与评估
- 开展实验并参与制定研究路线图
- 将研究成果交付并应用于训练运行
- 与 Anthropic 内外部的其他研究人员、工程师及性能工程专家协作
任职要求
- 具备加速器（CUDA、ROCm、Triton、Pallas）与机器学习框架编程（JAX 或 PyTorch）方面的专业知识
- 具有跨技术栈的工作经验，包括内核、模型代码与分布式系统
- 懂得平衡研究探索与工程实现
- 对 AI 的潜力充满热情，并致力于开发安全且有益的系统
- 具备强化学习经验者优先
- 具备在不同类型加速器之间移植机器学习工作负载的经验者优先
- 熟悉大语言模型训练方法者优先
- 最低学历要求：学士学位，或同等教育、培训及/或经验组合
- 所需专业：通过相关经历证明与岗位相关的领域