大模型系统训练研究员

职位描述

1. 参与深度学习框架的代码实现,开发框架支持大模型训练、微调能力，提升系统速度和系统稳定性。

2. 持续扩充核心框架的功能和计算能力,优化系统的架构和模型设计,保持一定的业务和技术前瞻性。

3.与产品和测试以及其他研发团队深入协作，解决关键问题，完成研发计划。

4.负责大模型训练相关技术，方案及架构设计沟通、规划、撰写。

5.调研分析海内外产品技术实现，探索创新方向。

任职要求

1.本科及以上学历、计算机、软件工程等相关专业优先。

2. 3年及以上相关工作经验，有扎实的计算机科学知识，熟练掌握Pytorch，具备良好的编程能力和代码风格。

3. 对AI大模型相关核心技术原理有完整的认识, 熟悉megatron deepspeed等种大模型加速框架。

4.有AI大模型性能调优经验，模型量化和稀疏化经验。

5.具备良好的技术、产品、项目文档撰写能力，调研能力、战略思维。

6.工作态度认真，做事主动、仔细。有强烈的责任心、有自驱力、抗压能力。

7.熟悉常见 AI 编程工具（如 Claude Code、Cursor、GitHub Copilot等），能将其有效融入日常开发工作流，具备借助 AI 工具进行代码编写、调试及提效的主动意识。

加分项:

1.有模型加速和压缩技术的实际项目经验，包括量化、剪枝、知识蒸馏等。

2.有高性能计算优化经验，了解cuda，TVM等。

3.熟悉RL的分布式训练框架，熟悉后后训练技术。