Xiaomi logo
小米
顶尖应届-AI Compiler 开发工程师-芯片

顶尖应届-AI Compiler 开发工程师-芯片

发布于 大约 15 小时前

普通员工/个人贡献者

北京市
无经验要求
全职员工
仅现场办公
博士
软件工程
Ai Compiler
Dsl
Gpgpu
Hpc
Llvm
Machine Learning
Npu
Simt
Tensor Core

AI 估算 · 30k–50k

博士应届顶尖人才,AI芯片方向热门,小米大厂薪资有竞争力,月薪3-5万合理

职位详情

关于这个职位

该职位是小米面向顶尖应届博士开设的AI编译器开发岗位,专注于端侧NPU的编译技术研究,包括自动向量化、计算切分、访存优化等前沿方向

你将参与从模型到硬件的端到端编译优化,利用AutoTuning和机器学习探索最佳配置,为AI芯片赋能
适合对编译器、体系结构有深厚研究背景的博士毕业生

最低要求

计算体系结构或者编译技术方向博士

从事GPGPU/NPU上AI编译技术研究,并且发表过顶级论文
有较强的分析和解决问题能力、较强的动手能力
熟悉LLVM等编译器,有传统编译器后端开发经验优先
熟悉GPGPU和SIMT架构,有NVGPU上编译优化,算子优化,HPC优化等经验优先
有开源社区贡献者优先

工作职责

结合端侧NPU和模型深入研究编译技术,例如针对计算单元的自动向量化,深度融合等

研究优化针对Tensor Core计算切分,多种并发模式优化等
研究访存优化,Layout最优化求解等
利用AutoTuning、机器学习等技术寻找最优化配置方案等
探索端到端模型编译新方法,Tile-based DSL设计与开发等

AI 洞察

优缺点分析

优点

  • 接触最前沿的AI芯片编译技术,与顶尖团队共同推动端侧推理性能极限
  • 小米作为大厂,资源丰富,业务落地场景真实,研究成果可直接产品化
  • 薪资待遇优厚,博士应届起薪具有市场竞争力,且有股票期权等长期激励
  • 团队技术氛围浓厚,鼓励创新和论文发表,有利于个人学术声誉积累
  • 技术门槛高,需要同时掌握体系结构、编译器、AI模型等多领域知识,学习曲线陡峭
  • 工作强度较大,处于芯片和AI交叉的快速演进领域,需要持续跟进新技术
  • 适合对编译器底层技术有浓厚兴趣、具备扎实体系结构背景、有志于在AI芯片领域深耕的顶尖博士毕业生

缺点 / 挑战

  • 研究导向型岗位,产出压力较大,需在工程实现和学术创新之间取得平衡

角色解读

  • 技术专家路线:深耕AI编译器领域,成为芯片编译技术带头人,主导关键优化技术
  • 架构师路线:转向系统级架构设计,统筹NPU、编译器与AI模型的协同优化
  • 学术+工业结合:持续发表顶级论文,代表公司在编译器顶会发声,提升技术影响力
  • 深入研究端侧NPU的编译器技术,包括自动向量化、深度融合等优化算法
  • 针对Tensor Core计算切分、并发模式优化、访存布局优化等具体问题进行算法设计与实现
  • 利用AutoTuning和机器学习方法自动搜索最优编译配置,提升模型推理效率
  • 探索端到端模型编译新范式,如Tile-based DSL的设计与开发,推动编译技术前沿
  • 扎实的计算机体系结构和编译原理背景,熟悉LLVM等编译器框架
  • 深入理解GPGPU和SIMT架构,有GPU/NPU上的编译或算子优化经验
  • 具备较强的算法分析与问题解决能力,能独立设计优化方案
  • 有顶级论文发表经验,熟悉学术研究方法与论文写作

申请策略

  • 面试前深入了解小米的端侧芯片产品线(如澎湃系列),思考如何将自身研究与产品结合
  • 准备一个完整的编译器优化案例,从问题定义、方案设计到实验对比的清晰讲述
  • 突出博士期间编译器或体系结构方向的核心研究成果,如发表的顶级论文、专利
  • 详细描述与GPGPU/NPU相关的优化项目,量化性能提升指标(如加速比、吞吐量)
  • 展示对LLVM等开源编译器的贡献经历,如提交的patch、参与的相关项目
  • 强调分析解决复杂问题的能力,例如在访存优化、自动向量化方面的具体案例
  • 若对NPU架构不熟悉,可提前学习寒武纪、昇腾等端侧NPU的编程模型
  • 强化AutoTuning和机器学习在编译器中的应用知识,如TVM、MLIR等框架

面试指南

  • 结构化回答:先给出结论,再分步骤或分层级阐述,最后总结
  • 例如优化问题:分析瓶颈→提出方案→实验验证→效果对比
  • 结合理论与实践:提及相关论文或开源框架作为支撑,同时给出自己的理解和改进思路
  • 展现深度思考:不仅要描述怎么做,还要解释为什么选择该方案,有无其他替代方案及其优劣
  • 请解释LLVM的pass架构,如何编写一个自定义的优化pass?
  • 针对NPU的架构特点,你会如何优化一个卷积算子?
  • AutoTuning在编译器中的应用有哪些?如何平衡搜索时间和优化效果?
  • 你博士期间解决过最复杂的编译优化问题是什么?具体步骤和结果如何?

职位点评

68
综合评分

小米顶尖应届AI编译器岗,技术前沿、薪资优厚,但工作强度大、WLB一般。

从起薪待遇、成长路径、工作节奏和岗位方向综合评估,方便比较职业起点。

更适合这类人
该职位最适合追求技术成长、乐于挑战前沿问题的博士毕业生,对工作生活平衡要求不高者优先。
表现最好
成长发展
相对薄弱
工作生活
薪资福利60
成长发展90
工作生活40
使命价值70

薪资福利

60中等

薪资处于市场高端水平,但JD未明确披露具体数字和福利,补偿性动机满足度中等偏上。

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

成长发展

90较高

职位涉及前沿编译技术,鼓励创新和论文发表,博士团队成长空间大,发展性动机满足度高。

技术前沿前沿/新兴技术
技术栈AI Compiler、LLVM、GPGPU、NPU、AutoTuning、Machine Learning、DSL
业务类型ambiguous

工作生活

40较低

仅现场办公,未提及弹性工作或WLB信号,大厂芯片部门通常工作强度较大,生活化动机满足度较低。

工作模式仅现场办公
办公地点市区核心地段
加班情况未提及(无法判断)

使命价值

70中等

AI芯片属于高速增长赛道,但JD未强调社会使命,意义感动机满足度中等。

行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
Watch Jobs