AI Infra实习生
岗位摘要
参与pretrain的性能优化与新模型结构探索工作,提升预训练探索效率;开发领先的RL Infra框架,支持最前沿的强化学习算法探索工作;提升LLM推理引擎的计算效率,实验并落地创新性推理方案,实现成本数量级下降
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 参与pretrain的性能优化与新模型结构探索工作,提升预训练探索效率
- 开发领先的RL Infra框架,支持最前沿的强化学习算法探索工作
- 提升LLM推理引擎的计算效率,实验并落地创新性推理方案,实现成本数量级下降
- 参与万卡规模训练与推理系统的效率优化工作
- 探索各类创造性解决思路,并在真实场景中精细打磨并落地实践
任职要求
- 熟练掌握Python、C++,擅长分析性能问题
- 熟练掌握CUDA,或拥有3年以上计算密集型优化经验
- 拥有分布式系统设计经验者优先
- ISC/ACM/NOI等竞赛获奖者优先
福利待遇
- 可接触最新的大模型研究工作
- 解决万卡规模的训练/推理效率问题
- 参与MoBA、Moonlight等前沿项目
- 实习待遇优厚
工作地址
北京海淀区京东科技大厦13层
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。