HirePortal

AI Infra专家/工程师

  • Xiaomi
  • Shanghai, Beijing, China
  • CNY 500,000 – CNY 800,000

北京、上海社招全职职位 ID:A154597

职位描述

  1. 基于高性能异构计算平台,设计开发面向自动驾驶场景的AI推理基础架构;
  2. 主导高性能推理框架的开发与优化,重点提升模型在复杂场景下的实时推理效率与能效比;
  3. 针对核心自驾模型进行计算图与算子级优化;
  4. 参与软硬件协同优化体系设计,为硬件架构升级提供算法侧优化建议;
  5. 协同算法团队推进模型在异构计算平台的性能调优与落地验证。

职位要求

  1. 计算机、电子、自动化等相关专业本科及以上学历,3年以上高性能计算或AI推理系统开发经验;
  2. 精通GPGPU编程(如CUDA、OpenCL)及高性能推理框架(如TensorRT/TensorRT-LLM、ONNX Runtime、TVM/MLC-LLM等)的开发与优化;
  3. 具备丰富的高性能算子开发与优化经验,熟悉主流AI模型(特别是视觉、多模态及扩散模型)的推理特性;
  4. 对自动驾驶场景有深刻理解,熟悉自驾场景模型的推理优化;
  5. 有异构计算平台(含GPU/FPGA/专用加速器等)开发经验者优先;
  6. 优秀的系统分析、解决问题和团队协作能力。

投递

Skills

  • CUDA
  • TensorRT
  • ONNX Runtime
  • TVM
  • GPGPU Programming
  • AI Inference Optimization
  • Heterogeneous computing

Related jobs

XiaomiApply for this job