【星火X计划】面向超大尺寸通用LLM的预训练Infra工程(J14507)

地点

合肥、北京

团队

词元星火

雇佣形式

全职

招聘类型

校招

尚未披露：薪资、雇主申请截止时间

投递在官方渠道完成，完整外部回执留在本人工作区；已登录时，通过 CLI 明确记录的选择、本人自报投递和渠道观察三类最小事件默认上报平台。浏览或保存岗位不会触发上报，外部投递不会出现在“我的申请”里。

• 本课题聚焦于构建和优化面向大模型预训练的高性能、高可用训练基础设施，为模型团队提供稳定、高效可扩展的训练工程底座。
• 参与大规模分布式训练框架的设计、迭代与优化，适配预训练以及后训练完整流程，联合平台保障千卡/万卡量级集群实现高效稳定训练；

任职要求与期待

• 24-27届硕士及以上学历，计算机、软件工程、高性能计算、电子信息、数学等相关专业
• 计算机体系结构、深度学习相关理论功底扎实
• 编程基础牢靠，熟练掌握Python，同时精通C++、CUDA二者其一，能够产出规范、高性能的工程代码
• 掌握如下技术/知识一种：主流LLM深度学习训练框架、GPU/国产芯片高性能算子编程、编译优化
• 掌握GPU架构特性以及CUDA开发手段，了解昇腾NPU、海光DCU等国产加速硬件配套软件栈，熟悉算子优化方法与FP16/BF16/FP8混合精度训练方案
• 良好的沟通协作能力，自驱力和责任心足够强，可以和团队共同攻坚并落地新技术

优先条件（非硬门槛）

• 学业排名、科研成果、硬核竞赛表现突出者优先