岗位描述(节选)
【岗位职责】
1.主导千卡/万卡级别大模型分布式训练系统的架构设计与实现
2.设计并优化 3D 并行策略(张量并行 + 流水线并行 + 数据并行)
3.解决大规模训练中的稳定性、效率、通信优化等问题
4.制定技术路线图,指导团队技术方向
5.与算法团队紧密协作,针对模型特性定制优化方案
【任职要求】
•5+ 年大规模分布式系统经验,有万卡集群训练经验优先
•深入理解 Megatron-LM、DeepSpeed、ColossalAI 等训练框架源码
•精通 NCCL/CCL 通信库,有通信优化经验
•熟悉 CUDA 编程,了解 GPU 硬件架构
•具备系统性能调优经验,MFU 优化实战案例
•良好的团队协作与跨团队沟通能力
加分项:
•顶会论文(SC、PPOPP、HPC 等)
•开源项目核心贡献者
•针对特定模型(GPT、LLaMA、Moe)的训练优化经验
本站当前在招 24732 个上海岗位(全城各职能合计),其中 1207 个公开标注了薪资。按这些标注统计:月薪(税前)主流区间 8K ~ 20K(25~75 分位),中位约 12K,最低 0.8K、最高 113K。
在招岗位较多的企业:字节跳动 954 个米哈游 691 个拼多多 497 个哔哩哔哩股份有限公司 452 个阿里巴巴集团 442 个
统计口径:求职者营地实时在招岗位的公开薪资标注,未标注者不计入;区间为 25~75 分位,非薪酬承诺。