小米 校园招聘
顶尖应届-AI Compiler 开发工程师-芯片
北京 ·经验不限·学历不限
北京
薪资面议
前往 小米 官网投递
职位描述
1、结合端侧NPU和模型深入研究编译技术,例如针对计算单元的自动向量化,深度融合等;
2、研究优化针对Tensor Core计算切分,多种并发模式优化等;
3、研究访存优化,Layout最优化求解等;
4、利用AutoTuning、机器学习等技术寻找最优化配置方案等;
5、探索端到端模型编译新方法,Tile-based DSL设计与开发等。
【任职要求】
1、计算体系结构或者编译技术方向博士;2、从事GPGPU/NPU上AI编译技术研究,并且发表过顶级论文;3、有较强的分析和解决问题能力、较强的动手能力;4、熟悉LLVM等编译器,有传统编译器后端开发经验优先;5、熟悉GPGPU和SIMT架构,有NVGPU上编译优化,算子优化,HPC优化等经验优先;6、有开源社区贡献者优先。
官网发布:2026-07-21 · 最后确认在招:2026-09-03 23:17:24 · 来源平台:xiaomi