小米 校园招聘
大模型推理优化工程师
北京 ·经验不限·学历不限
北京
薪资面议
前往 小米 官网投递
职位描述
1. 参与大模型推理优化和计算框架的设计与开发,探索提升推理效率的技术方案;
2. 学习并应用量化、并行计算、访存优化等方法,协助优化模型推理时延和吞吐;
3. 参与模型性能分析与调优实践,定位并解决性能瓶颈问题;
4. 参与自动化工具和流程的开发,简化模型推理部署流程;
5. 跟踪大模型推理领域最新研究进展,提出创新想法并推动落地。
【任职要求】
1. 计算机、数学、人工智能等相关专业本科及以上学历,2027届毕业生;
2. 扎实的编程基础,熟练掌握 C++/Python,有良好的代码风格和工程习惯;
3. 了解机器学习/深度学习基础知识,熟悉 PyTorch 等主流框架;
4. 对大模型推理优化方向有浓厚兴趣,了解 vLLM、SGLang 等推理框架者优先;
5. 有以下任一经验者优先:
- 参与过开源项目或推理框架社区贡献;
- 有 CUDA/GPU 编程、高性能计算相关实践经验(有github相关项目);
- 有模型量化(GPTQ/AWQ/FP8等)、算子优化、内存优化等方向的实践或研究经历;
- 有投机采样(Speculative Decoding)、Medusa/EAGLE/MTP/DFlash 等加速解码相关经验;
- 有强化学习(RLHF/GRPO/DPO)训练或推理相关经验;
- 具备学术论文阅读与复现能力;
6. 良好的学习能力、沟通能力和团队协作精神。
官网发布:2026-08-23 · 最后确认在招:2026-09-03 23:17:24 · 来源平台:xiaomi