开发者技术（训练）工程师
岗位摘要
深入客户训练项目，从算子、框架到算法进行全栈性能优化，攻坚MoE、强化学习、多模态等前沿工作负载；面向千卡/万卡集群优化分布式训练，包括并行策略、通信/显存/算子协同，提升大模型训练吞吐与算力利用率
岗位职责
- 深入客户训练项目，从算子、框架到算法进行全栈性能优化，攻坚MoE、强化学习、多模态等前沿工作负载
- 面向千卡/万卡集群优化分布式训练，包括并行策略、通信/显存/算子协同，提升大模型训练吞吐与算力利用率
- 将客户场景的性能瓶颈与技术洞察反馈至GPU架构与软件栈，驱动国产AI训练平台持续演进
任职要求
- 熟悉Python或C++，具备扎实的算法和计算机基础，编程风格良好
- 对并行计算有深入理解，具备敏锐的性能分析与调优能力
- 熟悉深度学习和大模型训练相关算法，熟练使用PyTorch/JAX/Paddle等主流框架
- 具备良好的沟通协作能力，能与架构、研究及客户团队高效推动优化落地
- 具有强烈的工作责任心与好奇心，学习能力强，自驱力高
- 善于使用AI/Agent工具辅助研发，持续跟进前沿技术动态
- 【加分项】有大规模GPU集群性能优化和训练经验者优先
- 【加分项】熟悉Megatron/DeepSpeed/ColossalAI/torchtitan/SGLang/vLLM等大模型训练或推理框架者优先
- 【加分项】熟悉大模型算法，如基础架构、强化学习、多模态、视频生成等方向者优先
- 【加分项】在顶级学术会议发表论文或有竞赛经验者优先
- 【加分项】熟悉CUDA/MUSA相关开发，有其他优秀国产GPU开发经验者优先
加分项
- 有大规模 GPU 集群性能优化和训练经验优先
- 熟悉大模型训练框架和推理框架，包括不限于 Megatron/DeepSpeed/ColossalAI/torchtitan/SGLang/vLLM 等
- 熟悉大模型算法，如大模型基础架构/强化学习/多模态/视频生成等
- 在顶级会议发表论文或有竞赛经验优先考虑
- 熟悉 CUDA/MUSA 相关开发，有其他优秀国产卡开发经验者优先
- 常先生 刚刚活跃
福利待遇
- 薪资范围：30,000–60,000元/月，共16薪
- 工作地点：北京朝阳区望京国际研发园
工作地址
北京朝阳区望京国际研发园-I座1