大模型系统训练研究员
岗位摘要
参与深度学习框架的代码实现，开发支持大模型训练和微调的能力，提升系统速度和稳定性；持续扩充核心框架的功能和计算能力，优化系统架构和模型设计，保持业务和技术前瞻性；与产品、测试及其他研发团队深入协作，解决关键问题，完成研发计划
岗位职责
- 参与深度学习框架的代码实现，开发支持大模型训练和微调的能力，提升系统速度和稳定性
- 持续扩充核心框架的功能和计算能力，优化系统架构和模型设计，保持业务和技术前瞻性
- 与产品、测试及其他研发团队深入协作，解决关键问题，完成研发计划
- 负责大模型训练相关技术、方案及架构设计的沟通、规划和撰写
- 调研分析海内外产品技术实现，探索创新方向
任职要求
- 本科及以上学历，计算机、软件工程等相关专业优先
- 3年及以上相关工作经验，有扎实的计算机科学知识，熟练掌握PyTorch，具备良好的编程能力和代码风格
- 对AI大模型相关核心技术原理有完整认识，熟悉Megatron、DeepSpeed等大模型加速框架
- 有AI大模型性能调优经验，以及模型量化和稀疏化经验
- 具备良好的技术、产品、项目文档撰写能力，以及调研能力和战略思维
- 工作态度认真，做事主动、仔细，有强烈的责任心、自驱力和抗压能力
- 熟悉常见AI编程工具（如Claude Code、Cursor、GitHub Copilot等），能将其有效融入日常开发工作流，具备借助AI工具进行代码编写、调试及提效的主动意识
加分项
- 有模型加速和压缩技术的实际项目经验，包括量化、剪枝、知识蒸馏等
- 有高性能计算优化经验，了解cuda，TVM等
- 熟悉RL的分布式训练框架，熟悉后后训练技术