岗位描述
工作职责
1、负责大语言模型预训练、强化学习训练框架研发与优化
2、与算法团队深度合作,进行算法与系统的联合设计与优化
3、保持关注行业前沿技术,且有能力和热情开展创新研究
任职要求
1、有人工智能、计算机系统等相关专业的研究经历
2、熟练掌握Linux环境下的C/C++与Python语言
3、精通以下至少一项的背景知识或经验:分布式训练框架、强化学习框架、高性能GPU计算和通信
4、具有独立解决问题的能力,良好的团队合作精神
5、有强烈的工作责任心,较好的学习能力、沟通能力和自驱精神
6、有良好的文档习惯,及时按要求撰写更新工作流程及技术文档
加分项
- 深入研究过至少一种主流分布式训练框架的底层架构和原理,或具备开发高性能CUDA kernel的经验
- 有硕士/博士阶段的计算机系统方向研究背景,作为核心作者在国际顶级AI会议上发表过论文;或深度参与贡献过开源代码项目
- 参与过LLM训练流程,有算法系统协同设计的经历