
小米
顶尖应届-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