Xiaomi logo
小米
AI软件工程师

AI软件工程师

发布于 大约 14 小时前

普通员工/个人贡献者

北京市 / 西安市
高级经验
全职员工
仅现场办公
硕士
机器学习工程
Android Nn
Pytorch
Rtos
Tensorflow
图编译器
芯片驱动
高性能算子

AI 估算 · 25k–50k

AI芯片软件工程师技术门槛高,小米平台加持,一线城市薪资可观,硕士+精通要求对应资深级别。

职位详情

关于这个职位

该职位负责小米自研芯片的AI软件栈开发,涵盖底层驱动、高性能算子、推理框架和图编译器等多个方向

你将参与端到端AI解决方案的交付,并针对自研指令集进行深度优化,涉及模型性能与功耗调优
适合对芯片软件、深度学习框架和底层系统开发有浓厚兴趣的工程师

最低要求

图像处理,模式识别,机器学习等相关专业硕士及以上学历

精通C&C++/Python软件编程语言,具有扎实的代码功底和交付能力
熟悉Android NN/TensorFlow/Pytorch中的一种神经网络框架
熟悉Android/Linux框架,具备Linux/RTOS等平台开发经验
良好的团队沟通能力,技术攻关能力

工作职责

参与芯片底层驱动的开发,方案设计和交付,完成AI多任务调度,Firmware开发,kernel驱动等

参与高性能算子的开发,基于自定义指令集开发高性能算子
参与AI推理框架的开发,方案设计,完成量化,图转换等工具链的开发以及图优化
参与AI 图编译器的开发,方案设计,基于自定义指令集完成图编译器的开发和交付,完成opfusion,图融合,片上内存分配,指令调度等
参与AI模型的性能优化,功耗优化
参与自研芯片的FPGA、EMU验证

AI 洞察

优缺点分析

优点

  • 身处AI芯片前沿赛道,技术含量高,能接触到从驱动到编译器的完整软件栈,个人技术成长空间大
  • 小米平台资源丰富,自研芯片战略投入大,岗位稳定性和发展前景较好
  • 工作内容涉及多个方向,可以拓宽技术视野,为未来职业发展提供更多可能性
  • 对底层系统知识和编程能力要求极高,工作强度可能较大,需要持续学习新技术
  • 自研芯片属于硬核研发,面临的技术难题多,调试和验证周期长,需要较强的耐心和抗压能力
  • 多地点办公可能涉及协作,有时需要跨团队沟通,增加沟通成本

缺点 / 挑战

  • 适合对AI底层技术有浓厚兴趣、具备扎实C++/Python功底、乐于挑战高性能计算和芯片软件栈的资深软件工程师

角色解读

  • 从专项技术模块(如驱动或算子)起步,逐步深入AI芯片全栈,成为芯片软件领域专家
  • 可向AI编译器/芯片架构方向纵深发展,或向技术管理岗位转型,带领团队完成更复杂的交付
  • 积累端侧AI落地经验,未来可转向边缘计算、自动驾驶等新兴AI领域
  • 负责AI芯片底层驱动的开发与设计,包括多任务调度、Firmware和kernel驱动,确保芯片高效运行
  • 基于自定义指令集开发高性能算子,对AI模型中的关键计算进行极致优化
  • 参与AI推理框架和图编译器的开发,完成量化、图转换、图融合、片上内存分配等工具链工作
  • 负责AI模型的性能与功耗优化,并参与自研芯片的FPGA和EMU验证
  • 精通C/C++和Python,具备扎实的代码功底和系统级软件开发能力
  • 熟悉至少一种主流神经网络框架(如TensorFlow、PyTorch)以及Android NN
  • 熟悉Android/Linux系统,具备Linux或RTOS平台上的开发经验
  • 良好的团队沟通能力和技术攻关能力,能解决复杂系统问题

申请策略

  • 关注小米招聘官网或内推渠道,了解具体岗位所属部门(如芯片部门),在申请时表达对AI芯片方向的热情
  • 面试中多展示你对端到端AI解决方案的理解,而不仅仅是单个模块的经验
  • 突出你在AI框架、编译器或芯片驱动方面的项目经验,尤其是性能优化相关的成果(如算子加速、推理延迟降低)
  • 强调C/C++和Python的熟练度,以及Linux/RTOS环境下的开发经验,可以附上代码仓库或技术博客
  • 如果有神经网络模型量化、图优化或算子实现的经验,务必重点展示
  • 体现团队协作和技术攻关的能力,如主导过复杂技术问题的解决
  • 提前熟悉AI编译器的基本概念(如TVM、MLIR)和GPU/NPU的算子优化方法,会给面试加分
  • 复习计算机体系结构、操作系统(中断、内存管理)和并行计算相关知识,以应对底层技术问题

面试指南

  • 采用STAR法则:背景-任务-行动-结果,突出你的技术决策和量化成果
  • 对于技术问题,先阐述原理,再结合项目实例说明具体实现,最后提出改进思路
  • 遇到不确定的问题,诚实说明边界,并展示你的分析过程和解决问题的能力
  • 请介绍一个你参与过的AI模型部署或性能优化项目,具体做了什么,效果如何?
  • 在Linux/RTOS环境下进行驱动开发时,你是如何调试和定位问题的?
  • 针对自定义指令集,你会如何设计并优化一个算子?请举例说明
  • 你熟悉哪些图优化技术?如何对计算图进行融合或内存分配优化?
  • 如何在资源受限的端侧设备上平衡模型性能和功耗?

职位点评

68
综合评分

小米自研芯片AI软件工程师,前沿技术、强成长,但工作强度与福利未知。

从薪资福利、成长空间、工作节奏和岗位方向综合评估,方便横向比较。

更适合这类人
适合以技术成长为首要目标的求职者,愿意投入高强度工作换取稀缺的AI芯片全栈经验。
表现最好
成长发展
相对薄弱
工作生活
薪资福利55
成长发展92
工作生活50
使命价值78

薪资福利

55较低

职位未披露具体薪资和福利信息,但作为上市大厂的核心技术岗位,预计整体薪酬具有竞争力,但稳定性与福利需在面试中确认。

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

成长发展

92较高

该职位深度参与AI芯片软件栈的多个前沿方向,技术成长空间巨大,能积累稀缺的自研芯片开发经验。

技术前沿前沿/新兴技术
技术栈C++、Python、TensorFlow、PyTorch、Android NN、Linux、RTOS、高性能算子、图编译器、FPGA、EMU
业务类型ambiguous

工作生活

50较低

JD未提及工作模式、加班情况或办公地点细节,仅能确定为线下办公,工作节奏未知,可能面临较高强度。

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

使命价值

78中等

AI芯片属于国家重点支持的硬科技领域,对国产芯片发展有积极意义,但JD未明确社会价值导向。

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