HirePortal

大模型部署优化工程师

  • Xiaomi
  • Shanghai, Beijing +1 more, China
  • CNY 400,000 – CNY 600,000

新业务部-大模型部署优化工程师

北京、上海、西安社招全职职位 ID:E9581

职位描述

  1. 参与Android 平台的大模型(aigc + llm)开发和部署;
  2. 负责大模型算子在NPU/GPU/CPU部署及高性能算子实现;
  3. 负责大模型异构部署的算子性能调优及优化工作;
  4. 分析算子/网络的性能瓶颈,提出性能优化策略,达成性能目标。

职位要求

  1. 本科及以上学历,计算机科学、电子工程或自动化等相关专业;
  2. 熟悉并行/异构计算体系结构,在TPU/NPU/GPU一种平台有3年以上的开发和优化经验;
  3. 熟悉计算机系统体系架构,软件性能优化加速;
  4. 熟练使用至少一种主流Al训练框架,如TensorFlow、PyTorch、Caffe等; 5.具备业界常用高性能库(TensorRT/cuDNN/CUDA/OPENCV)开发和性能优化经验者优先;
  5. 良好的团队沟通能力、技术攻关能力,责任心强。

投递

Skills

  • Large Language Model Deployment
  • NPU/GPU/CPU Optimization
  • CUDA
  • TensorRT
  • PyTorch
  • Performance Tuning
  • Heterogeneous computing

Related jobs

XiaomiApply for this job