直招.cv

← 返回职位列表

小米 社会招聘

新业务部-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中一种神经网络框架;

官网发布:2025-08-05 · 最后确认在招:2026-09-03 23:18:29 · 来源平台:xiaomi