Cambricon logo
寒武纪
AI编译器开发实习生-27届

AI编译器开发实习生-27届

发布于 大约 2 小时前

实习/见习

上海市
无经验要求
实习生
仅现场办公
学历未注明
实习与临时职位
Linalg
Mhlo
Mlir
Torchinductor
Tvm
Xla
编译器优化

AI 估算 · 5k–8k

AI芯片公司实习,技术含量高,薪资具竞争力

职位详情

关于这个职位

该职位是寒武纪AI编译器开发实习生,专注于优化Triton和MHLO编译器,使其适配寒武纪芯片架构,提升AI计算性能和效率

你将参与图优化、后端代码生成、算子融合等关键工作,与硬件和AI框架团队紧密协作
适合对AI编译技术有浓厚兴趣、具备C++/Python基础的在校学生

最低要求

l 优秀的学习能力,对 AI 编译器、AI计算优化有浓厚兴趣

l 熟悉 C++/Python
l 计算机体系结构基础知识
l 算子优化基础知识

工作职责

l 研究和优化 Triton 编译器和基于MHLO的图编译器,使其高效支持 寒武纪芯片 计算架构

l 基于 MLIR + Linalg 框架,提升编译器代码执行效率
l 参与 图优化、后端代码生成、调度策略、指令优化、流水并行化 相关优化工作
l 研究和优化 Kernel 计算性能,降低计算开销,提高吞吐率
l 研究算子融合(Op Fusion)、自动调优(Auto-Tuning)、代码生成(CodeGen) 相关优化策略
l 与 硬件团队、AI框架团队(如 PyTorch/TensorFlow) 协作,提升AI编译器的支持能力
l 跟踪 LLVM、Triton、TVM、XLA、TorchInductor 等 AI 编译器前沿技术,优化编译性能

优先资格

l 了解Triton、MHLO 等编译流程

l 了解ai编译优化(如 TVM、XLA、TorchInductor)
l 参与过 Triton、MLIR、IREE、XLA、TVM、TorchInductor 等编译器的开发,或在 GitHub 提交过相关 PR
l 具备MLU/NPU/GPU 相关的编译优化经验
l 有深度加速器上(gpu/npu)相关算子开发经验
l 在 NeurIPS、ICLR、PLDI、CGO、MICRO 等学术会议上发表过 AI 编译相关论文

AI 洞察

优缺点分析

优点

  • 寒武纪是AI芯片领域上市公司,实习经历含金量高,对简历加成显著
  • 接触Triton、MLIR等前沿编译器技术,紧跟AI编译发展趋势
  • 团队协作紧密,能快速积累工程实践和沟通协作能力
  • 实习周期长(≥6个月),有机会深度参与项目,产出实质成果
  • AI编译技术迭代快,需要持续学习和跟踪最新进展
  • 适合对AI编译优化有浓厚兴趣、具备较强编程能力和学习自驱力的计算机相关专业学生

缺点 / 挑战

  • 编译器开发难度较高,需要扎实的计算机体系和编译原理基础
  • 实习周期要求6个月以上,需平衡学业和工作压力

角色解读

  • 实习转正,成为寒武纪编译器团队正式工程师,深入参与核心产品研发
  • 深耕AI编译优化方向,逐步成为编译器技术专家,主导关键优化模块
  • 可向AI框架、芯片软件栈或高性能计算方向拓展,职业路径宽广
  • 优化Triton和MHLO编译器,使其高效适配寒武纪芯片架构
  • 基于MLIR+Linalg框架,提升编译器代码执行效率和性能
  • 参与图优化、后端代码生成、调度策略等编译优化工作
  • 与硬件团队和AI框架团队协作,提升编译器整体支持能力
  • 精通C++和Python编程,具备扎实的编码能力
  • 掌握计算机体系结构和算子优化的基础知识
  • 了解AI编译器框架如Triton、MLIR、TVM、XLA等
  • 熟悉编译原理、中间表示和代码生成相关技术

申请策略

  • 在简历和面试中强调对AI编译的兴趣和热情,展示主动学习能力
  • 了解寒武纪芯片架构(MLU)及AI编译器产品线,体现对公司的认同
  • 突出C++/Python编程项目经验,尤其是性能优化或底层开发相关
  • 展示编译器、体系结构或算子优化相关的课程作业或研究经历
  • 如有开源贡献(如向Triton、MLIR等提PR),务必列出并说明贡献点
  • 附加个人博客或技术笔记,体现对编译技术的热情和深度
  • 提前学习Triton、MLIR、Linalg等框架的基本概念和使用方法
  • 复习编译原理知识,尤其是中间表示、优化pass和代码生成

面试指南

  • 针对技术概念题:先给出定义,再结合具体例子说明,最后联系寒武纪业务
  • 针对项目经历题:采用STAR法则,突出挑战、思路、实施和结果
  • 针对开放性问题:展现对技术的热情和思考深度,提出个人见解
  • 请简述Triton编译器的基本流程和主要组成部分
  • MLIR中Dialect和Pass的概念是什么?你如何理解?
  • 描述一次你优化算子的经历,用了哪些方法和工具?
  • 计算机体系结构中,内存层次如何影响计算性能?
  • 你如何看待AI编译器未来的发展趋势?

职位点评

71
综合评分

AI芯片公司实习,前沿编译器技术,成长空间大,但薪资福利未明确。

从学习成长、工作节奏、岗位方向和实习待遇综合评估,方便比较实习机会。

更适合这类人
适合高度重视技术成长、对AI编译有热情、并能接受实习薪资不确定性的学生。
表现最好
成长发展
相对薄弱
薪资福利
薪资福利50
成长发展90
工作生活50
使命价值85

薪资福利

50较低

实习薪资未披露,福利不明确,但寒武纪作为上市公司提供基本保障。

薪资信号未披露(AI估算:5K-8K/月)

成长发展

90较高

接触前沿AI编译器技术,参与核心优化,成长空间大。

技术前沿前沿/新兴技术
技术栈C++、Python、Triton、MLIR、Linalg、LLVM、TVM、XLA
成长机会实习
业务类型ambiguous

工作生活

50较低

仅现场办公,工作地点未明确,WLB信号不足。

工作模式仅现场办公
办公地点未明确
加班情况未提及(无法判断)

使命价值

85较高

AI芯片自主可控,推动国产AI基础设施发展,社会价值较高。

行业发展高速增长赛道
社会影响正向社会影响力较高
创新程度积极采用新技术
Watch Jobs