返回岗位列表

摩尔线程

AI端侧推理优化工程师

地点:北京 薪资:40-60K 日期:2026-03-03

岗位摘要

负责vLLM、llama.cpp、MNN等端侧大模型推理框架的性能优化与加速,提升吞吐并降低延迟;负责语音、Embedding、OCR、AIGC、自动驾驶等模型在ONNXRuntime、MNN等框架中的性能调优与部署适配

岗位职责

  • 负责vLLM、llama.cpp、MNN等端侧大模型推理框架的性能优化与加速,提升吞吐并降低延迟
  • 负责语音、Embedding、OCR、AIGC、自动驾驶等模型在ONNXRuntime、MNN等框架中的性能调优与部署适配
  • 与算子、编译器团队紧密协作,完成端侧高性能推理框架的集成、验证与交付
  • 深入分析推理链路性能瓶颈,设计并实现系统级优化方案,保障框架在真实场景高效稳定运行

任职要求

  • 精通vLLM、llama.cpp、ONNXRuntime等至少一种主流推理框架的架构、工作机制与源码
  • 熟练掌握Flash Attention、Paged Attention、Speculative Decoding、Continuous Batching等大模型推理加速技术,并有实际应用经验
  • 熟练使用Triton、Cutlass开发高性能GPU Kernel,精通CUDA编程、内存管理及性能优化,有重要项目落地经验
  • 熟练使用PyTorch Profiler、Nsight Systems、Nsight Compute等性能剖析工具,掌握CUDA Graph、Torch AOT等高级优化技术
  • 责任心强,具备优秀沟通协作能力,能协同算法、编译、硬件等多团队完成端到端技术交付
  • 有一线互联网公司或头部AI企业推理引擎优化与大规模业务落地经验优先
  • 深入掌握算子优化、编译器优化或GPU硬件架构知识者优先
  • 具备端侧模型量化调优及部署实践经验者优先

加分项

  • 有一线互联网公司或头部AI企业推理引擎优化与大规模业务落地经验
  • 在推理框架技术基础上,深入掌握算子优化、编译器优化(如MLIR、TVM)或GPU硬件架构知识
  • 具备端侧模型量化(如W4A16、W4A8等)调优及部署实践经验
  • 胡先生 刚刚活跃

福利待遇

  • 薪酬区间40-60K,具备市场竞争力
  • 工作地点位于北京望京国际研发园,交通便利
  • 团队技术氛围浓厚,与顶尖专家共事
  • 参与前沿AI推理框架研发,技术成长空间大

工作地址

北京朝阳区望京国际研发园I座