返回岗位列表

大模型训练并行计算工程师

查看原岗位

地点：北京薪资：40-70K 日期：2026-03-03

岗位摘要

负责大模型训练的系统性能分析及解决方案制定，高效分析业务瓶颈；跟进和研究大模型训练性能和内存的优化手段，提升大模型训练效率；跟进业界大模型训练的前沿工作，对MoE、长序列、文生图等场景有深刻理解

岗位职责

负责大模型训练的系统性能分析及解决方案制定，高效分析业务瓶颈
跟进和研究大模型训练性能和内存的优化手段，提升大模型训练效率
跟进业界大模型训练的前沿工作，对MoE、长序列、文生图等场景有深刻理解
熟练掌握并行策略的通信计算模式，根据业务特点不断优化分析已有并行策略
熟悉大模型训练整体流程和评价指标，完成监控、稳定性训练等业务支持和开发
设计和建设大模型训练的性能内存评估方法，分析性能原因、通信比例、内存占用及计算资源消耗

任职要求

本科及以上学历，计算机相关专业
熟悉大模型训练框架Megatron-LM/Deepspeed等
具备良好的编程能力，熟练掌握性能分析工具和方法
熟悉PyTorch的性能分析工具
逻辑思维清晰，具有良好的沟通能力、抗压能力和责任心
具有实际开发或优化并行策略的经验优先
加分项：具备千卡以上集群训练经验

加分项

胡先生刚刚活跃

工作地址

北京海淀区北航致真大厦D座16层