返回岗位列表

小米

AI推理框架开发工程师

地点:北京 薪资:薪资未公开 日期:2026-01-29

岗位摘要

基于AI芯片完成AI推理框架的开发和交付,适配CPU/GPU/NPU多后端,支持CV/NLP/大模型等端侧落地;负责框架与AndroidNN/Tensorflow/Pytorch/Onnx等FrameWork的对接

岗位职责

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

任职要求

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