返回岗位列表

小米

AIGC大模型推理加速工程师

地点:北京 薪资:薪资未公开 日期:2026-07-20

岗位摘要

负责图像/视频/多模态等方向AIGC大模型推理加速研究,包括模型蒸馏、模型量化、模型剪枝、推理加速框架移植等;实现高性能LLM/VLM/AIGC模型推理引擎,优化计算图编译、动态Batching及KV Cache管理,打造极致低延迟与高吞吐的线上服务架构

岗位职责

  • 负责图像/视频/多模态等方向AIGC大模型推理加速研究,包括模型蒸馏、模型量化、模型剪枝、推理加速框架移植等
  • 实现高性能LLM/VLM/AIGC模型推理引擎,优化计算图编译、动态Batching及KV Cache管理,打造极致低延迟与高吞吐的线上服务架构
  • 持续跟进业界最新的Infra技术(如FlashAttention演进、高效通信库、FP8/INT8量化、稀疏蒸馏等训练部署方案),配合实际业务驱动技术预研并转化为生产力

任职要求

  • 本科及以上学历,计算机相关专业优先,熟练掌握C++/Python,精通数据结构与算法,并有并发编程基础
  • 熟练使用PyTorch/TensorFlow或其他深度学习框架,熟练掌握深度学习性能加速技术包括但不限于算子性能优化、通信优化等
  • 具有AIGC(VLM、Diffusion)模型训练/推理优化经验,熟悉TensorRT-LLM、vLLM、LightX2V等推理效率优化工具
  • 熟悉大模型蒸馏、量化、剪枝等常用压缩技术,熟练使用常见推理加速工具,如TensorRT、vLLM等
  • 积极主动有热情,具备较强的沟通能力以及优秀的团队协作能力