职位描述:
岗位职责:
1.进行大模型、多模态大模型、AIGC模型、自驾模型等多个方向端侧加速优化算法的研发与探索
2.工作内容包括但不限于上述模型的训练调优、量化、剪枝、小型化、稀疏优化、投机推理、专用加速、结构搜索(NAS)等
3.与芯片团队配合针对相关加速特性进行芯片算法联合设计与优化,并提供针对自研芯片的完整算法加速解决方案
4.帮助客户在自研芯片上进行算法加速优化的模型加速优化训练及交付
职位要求:
岗位要求:
1.从事大模型、AIGC、深度学习、AI芯片等两个或以上相关领域工作经验,有2年及以上的工作经验,主导或者作为主要参与者;
2.对AI算法(尤其是大模型、AIGC、AD算法)的量化剪枝、网络结构搜索、稀疏化、性能效果优化和训练调参等有丰富开发调试经验;
3.精通transfomers,diffuser,pytorch,tensorrt等框架或平台;
4.参加过完整芯片算法联合优化特性的设计和商用落地者优先;
5.有大模型/AIGC/AD算法加速优化项目经验者或在相关领域发表过学术论文或有开源项目者优先;
招聘部门:
小米
工作地点:
北京市社招全职职位 ID:A7018
面试建议:
这个岗位是小米在AI前沿领域的核心攻坚职位,最大的特点是要求算法工程师具备芯片级的优化思维。不同于普通的大模型算法岗,你需要同时掌握算法层面的压缩优化技术(如量化剪枝)和硬件层面的加速特性适配能力,特别是要能与自研芯片团队进行联合设计。 建议重点准备三个维度的能力证明:首先是用具体案例展示你对transformer/diffuser等框架的深度改造经验,最好是能带来显著推理速度提升的优化方案;其次要准备芯片协同设计相关的项目细节,比如如何根据芯片特性调整算法结构;最后要突出商用落地思维,面试官会特别关注你能否将学术成果转化为实际产品加速方案。记得带上能直观展示优化效果的对比数据,比如某模型经过你的优化后延迟降低百分比等硬指标。