返回岗位列表

摩尔线程

大模型推理训练优化工程师

地点:北京 薪资:200-300元/天 日期:2026-05-08

岗位摘要

PyTorch/Paddle模型适配与性能调优;多机多卡大模型训练部署与性能调优;vllm/SGLang以及推理引擎部署与性能调优;开发高性能算子;对社区CUDA仓库进行musify;调研最新的机器学习技术和平台功能

岗位职责

  • PyTorch/Paddle模型适配与性能调优
  • 多机多卡大模型训练部署与性能调优
  • vllm/SGLang以及推理引擎部署与性能调优
  • 开发高性能算子
  • 对社区CUDA仓库进行musify
  • 调研最新的机器学习技术和平台功能
  • 深度参与研发产品特性开发,为产品支持新特性

任职要求

  • 积极乐观,责任心强,工作认真细致,具有良好的团队沟通与协作能力
  • 计算机科学、电子工程或相关领域的本科及以上学历
  • 熟悉机器学习和深度学习算法及其在实际应用中的部署
  • 至少有一种机器学习框架使用或开发经验,如TensorFlow、PyTorch、PaddlePaddle、Megatron-LM、Colossal-ai、DeepSpeed、vLLM、SGLang或其他推理引擎
  • 熟悉至少一种编程语言,如CUDA、C++、Triton、Python
  • 良好的逻辑思维能力,精通计算机数据结构和算法

福利待遇

  • 薪资待遇优厚,日薪200-300元
  • 参与前沿AI大模型训练与推理优化项目
  • 探索GPU性能极限,积累行业领先技术经验
  • 团队氛围开放,沟通协作顺畅

工作地址

北京朝阳区望京国际研发园l座3层