
小米
AI 模型加速优化算法工程师
AI 模型加速优化算法工程师
发布于 大约 9 小时前普通员工/个人贡献者
北京市
中级经验
全职员工
仅现场办公
本科
机器学习工程
Diffusion
Onnx Runtime
Pytorch
Tensorrt
Tvm
推荐系统
模型压缩
模型量化
自动驾驶
AI 估算 · 30k–60k
小米属互联网大厂,AI算法岗薪资较高,北京地区硕士/博士起薪约30-60K,考虑股票和年终奖,综合月薪中位数45K。
职位详情
关于这个职位
作为AI模型加速优化算法工程师,你将专注于大模型、多模态模型和AIGC等方向在端侧的加速优化,负责模型压缩、量化、剪枝等技术的研发,并与芯片团队协同进行算法-芯片联合设计,推动自研芯片上的模型高效部署
该岗位深度参与前沿AI技术在移动端的落地,技术挑战大,成长空间广阔
最低要求
在大模型、AIGC、自动驾驶、多媒体/ISP、AI 芯片等方向上具有相关项目或实习经验
熟悉模型量化、剪枝、蒸馏、稀疏化、训练调优及性能优化等模型压缩与加速技术
熟练使用 PyTorch,熟悉 Transformer、Diffusion 等模型,掌握 TensorRT、ONNX Runtime、TVM 等至少一种部署或优化框架
工作职责
负责大模型、多模态大模型、AIGC、自动驾驶、多媒体/ISP 等 AI 模型端侧加速优化算法的研发与探索
负责模型训练调优及压缩优化,包括但不限于量化、剪枝、蒸馏、稀疏化、小型化、NAS、推理加速等技术研发
与芯片团队协同开展算法-芯片联合设计与优化,提供面向自研芯片的模型加速解决方案
支持业务模型在自研芯片上的训练优化、部署交付及性能持续优化
优先资格
有芯片算法联合优化、端侧 AI 部署或模型加速项目经验者优先
有大模型/AIGC/自动驾驶/多媒体 ISP 模型优化经验,或相关论文、专利、开源项目者优先
AI 洞察
优缺点分析
优点
- 小米有自研芯片和庞大端侧设备,技术落地场景丰富,个人影响力大
- 涉及大模型、AIGC、自动驾驶等前沿方向,技术成长快,行业前景好
- 大厂平台,薪资福利有竞争力,且有专利、论文等成果产出机会
- 算法-芯片联合优化跨领域知识要求高,需要较强的学习能力和系统思维
缺点 / 挑战
- 工作强度较大,需紧跟AI前沿技术,持续学习压力高
- 端侧资源受限,优化难度大,可能面临较大的技术攻关压力
- 适合对AI模型优化和底层实现有浓厚兴趣,具备扎实深度学习基础,喜欢挑战技术难题并愿意在端侧AI方向深耕的工程师
角色解读
- 初期聚焦模型压缩与加速技术,成为端侧AI优化专家
- 中期可向算法-芯片协同设计方向深入,参与自研芯片架构定义
- 长期可发展为公司AI基础设施技术负责人,或转向AI平台/产品化方向
- 研发大模型、多模态模型在手机等端侧的加速优化算法,包括量化、剪枝、蒸馏等压缩技术
- 与芯片团队合作,进行算法与芯片的联合设计,提升自研芯片上的模型推理效率
- 负责模型从训练到部署的全链路性能优化,确保AI能力在端侧高效运行
- 扎实的深度学习基础,熟悉常见模型结构如Transformer、Diffusion
- 熟练掌握PyTorch,并熟悉至少一种推理优化框架(TensorRT、ONNX Runtime、TVM)
- 了解模型压缩技术(量化、剪枝、蒸馏等),具备实际项目经验者优先
- 对硬件架构有一定理解,能够进行算法与芯片的协同优化
申请策略
- 建议在简历中明确表述对端侧部署的深刻理解和热忱,并展示一项端侧优化的完整案例
- 了解小米在AIoT、汽车等领域的产品布局,在面试中结合业务场景谈优化思路
- 突出模型压缩或加速相关项目,如量化、剪枝、蒸馏的实际落地效果
- 强调对PyTorch等框架的熟练度,以及使用TensorRT/TVM等优化工具的具体经验
- 如果有芯片相关或端侧部署经验,务必突出,这是加分项
- 列出相关论文、专利或开源项目,体现技术深度
- 深入学习TensorRT/TVM的底层优化原理,动手实践常用优化技巧
- 了解硬件架构(如NPU、GPU)基本原理,通过公开资料或课程补充知识
面试指南
- 回答技术问题采用STAR法则:情境(Situation)、任务(Task)、行动(Action)、结果(Result),突出数据指标和权衡
- 对于优化类问题,可按“分析瓶颈—提出方案—实验验证—迭代优化”的流程阐述,体现系统性思维
- 涉及硬件结合的问题,要体现对计算、内存带宽等基础概念的掌握,并强调软硬件协同的思想
- 请介绍一次你进行模型压缩的经历,如何平衡精度与速度?
- 针对Transformer模型,你会如何设计端侧加速方案?
- 谈谈你对量化中PTQ和QAT的理解及适用场景
- 如何评估一个端侧推理引擎的性能?你会关注哪些指标?
- 如果让你优化一个Diffusion模型在手机上的推理,你会从哪些方面入手?
职位点评
70
综合评分
前沿技术挑战大,薪资竞争力强,工作强度可能较大,WLB一般。
从薪资福利、成长空间、工作节奏和岗位方向综合评估,方便横向比较。
更适合这类人
适合追求技术前沿与快速成长,对薪资和WLB要求相对灵活的求职者。
表现最好
成长发展
相对薄弱
工作生活
薪资福利75
成长发展85
工作生活40
使命价值70
薪资福利
75中等
小米作为大型上市企业,提供具有竞争力的薪资福利,工作稳定,但具体薪酬未在JD中披露。
薪资信号未披露(AI估算:30K-60K/月)
成长发展
85较高
职位涉及大模型、AIGC等前沿技术,且与自研芯片结合,技术挑战大,成长空间广阔,能显著提升核心竞争力。
技术前沿前沿/新兴技术
技术栈PyTorch、Transformer、Diffusion、TensorRT、ONNX Runtime、TVM、量化、剪枝
业务类型profit_center
工作生活
40较低
JD未提及远程办公或弹性工作安排,通常为现场办公,互联网大厂节奏可能较快,WLB一般。
工作模式仅现场办公
办公地点市区核心地段
加班情况未提及(无法判断)
使命价值
70中等
AI模型加速有助于让AI更高效运行在硬件设备上,推动AI技术普及,具有一定社会价值,且行业前景好。
行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
小米 的其他在招职位
相似职位推荐
Watch Jobs