拓达教育 Touchdown Education 拓达求职支持办公室Career Support Office touchdown.org.cn ↗
JobFinder为留学生筛选身份匹配的岗位

大模型训练优化工程师-27届

寒武纪 · 北京 · 上海 · 收录于 8月15日
评分
80
发布日期
收录于 8月15日 (2周前)
内推
暂无
公司规模
1k-5k
行业
半导体

岗位描述

岗位职责 1、负责SOTA AI模型在寒武纪人工智能芯片(MLU)上的端到端性能优化和精度调优,沉淀可复用、可泛化的优化方法论,独立开发寒武纪性能、精度分析工具,支撑常态化优化工作。 2、负责寒武纪Model Zoo的开发、迭代与日常维护,基于MLU芯片完成各类SOTA大模型、多模态算法的完整复现,针对模型训练全流程做性能分析、精度调优,输出标准化的行业领域参考解决方案。 3、深入研究大模型底层核心技术,深耕Megatron、DeepSpeed等分布式训练框架,掌握剪枝、量化训练等模型加速技术,熟悉VERL、Swift、PEFT等主流训练工具链,定位并解决大模型训练过程中通信、计算、显存、调度等核心问题。 4、跟踪RLHF、MOE、各类attention、Agentic RL、OPD等大模型前沿技术方向,完成技术调研、原理验证与落地适配,将前沿技术成果转化为寒武纪自有训练框架的功能迭代,丰富芯片生态能力。 任职资格 1、电子信息、计算机、软件工程、自动化等EECS相关专业,本科及以上2026届应届毕业生,专业基础扎实。 2、熟悉Megatron、DeepSpeed、torch FSDP等主流大模型分布式训练框架,具备实验室项目、竞赛、科研项目等大规模分布式训练实操经验。 3、在大模型预训练、后训练、多模态、分布式并行等任一方向有项目积累、科研实践或独立研究见解。 4、具备GPU模型性能优化、精度调优实战经验,熟练使用nvprof、nvvp、Nsight等性能分析工具,能够独立完成模型瓶颈定位与优化。 5、精通PyTorch、TensorFlow等至少一种主流深度学习框架,熟悉框架底层运行逻辑与基础调优方法。 注:本岗位设置笔试环节,请留意邮件或短信通知、

加入我的投递 →

寒武纪 的其他岗位

大模型训练系统实习生-27届 · 北京 · 评分 86
大模型和分布式推理优化实习生(日常实习) · 上海 · 评分 84
大模型训练系统实习生(日常实习) · 北京 · 评分 84
AI编译器研发工程师-27届 · 北京 · 上海 · 评分 80
AI 智算平台研发工程师-27届 · 上海 · 合肥 · 评分 78

查看全部 →