小米 – 新业务部-AI推理框架开发工程师/专家 职位分析和面试指导

职位描述:

基于AI芯片,完成AI推理框架的方案设计及开发,包括:
1.基于AI芯片,完成 AI 推理框架的开发和交付,适配CPU/GPU/NPU多后端,支持CV/NLP/大模型等端侧落地;
2.负责框架与FrameWork的对接,包括AndroidNN/Tensorflow/Pytorch/Onnx等;
3.负责框架相关工具链的开发,包括Memory/PerformanceProfiling,量化工具等;
4.负责CV/NLP/大模型等算法业务的端到端性能优化,满足商用交付需求

职位要求:

1.硕士及以上学历,计算机/电子/信息/通信/自动化/软件等相关专业,有AI工程优化经验的优先;
2.精通C/C++,精通算法与数据结构,熟悉Python;
3.深入了解PADDLE/MNN/TVM/MLIR等框架中的一种或多种,熟悉算子性能优化、算子融合、graph遍历等技术;
4.熟悉NPU或其他AI芯片工具链开发, 熟悉模型量化技术优先;
5.了解主流 AIGC 算法模型原理,有AIGC模型加速优化经验者优先。
6.熟悉Android NN/TensorFLow/Pytorch/Caffe中一种神经网络框架;

招聘部门:

小米

工作地点:

上海市、北京市社招全职职位 ID:K9595

面试建议:

这个AI推理框架开发岗位对技术深度和广度都有很高要求。小米作为科技巨头,在AI领域布局深远,这个岗位不仅需要扎实的编程基础,更需要深入理解AI推理框架的底层实现和优化技巧。特别值得注意的是,职位要求候选人能够适配多种后端,这意味着你需要对不同硬件架构有深入理解。同时,大模型和AIGC相关的优化经验也是加分项,这反映了行业最新的技术趋势。 在准备面试时,建议你重点复习C++和Python编程,特别是与AI框架相关的代码实现。准备好展示你在框架优化方面的实际经验,比如算子融合、graph遍历等具体技术。对于芯片工具链和量化技术也要有所准备,这些都是面试官可能重点考察的内容。此外,小米很看重商用交付能力,所以可以准备一些你过去项目中性能优化的具体案例,用数据说明你的贡献。记得要展示你对最新AI技术趋势的理解,特别是大模型和AIGC相关经验。