返回岗位列表

小米

AI算法优化工程师

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

岗位摘要

进行多模态大模型、AIGC模型、感知模型、图像画质模型等算法端侧加速优化算法的优化加速与部署;模型的训练调优、量化(PTQ/QAT)、剪枝、小型化、稀疏优化、投机推理、专用加速、结构搜索(NAS)等

岗位职责

  • 进行多模态大模型、AIGC模型、感知模型、图像画质模型等算法端侧加速优化算法的优化加速与部署
  • 模型的训练调优、量化(PTQ/QAT)、剪枝、小型化、稀疏优化、投机推理、专用加速、结构搜索(NAS)等
  • 模型的优化部署:瓶颈分析,异构调度,自研芯片的定向优化等
  • 帮助算法在自研芯片上进行模型加速优化、部署及交付

任职要求

  • 计算机科学、电子信息、模式识别、计算机视觉、人工智能或相关专业硕士或博士学历
  • 对AI算法(尤其是大模型、AIGC、感知算法、图像画质算法)的量化剪枝、网络结构搜索、稀疏化、性能效果优化和训练调参等有丰富开发调试经验
  • 精通transfomers,diffuser,pytorch,tensorrt等框架或平台
  • 参加过完整芯片算法联合优化特性的设计和商用落地者优先
  • 有异构平台如GPU\NPU\DSP的算法加速优化的经验者优先
  • 有大模型/AIGC/AD算法加速优化项目经验者或在相关领域发表过学术论文或有开源项目者优先