AI算子库及高性能计算工程师(校招)
岗位摘要
基于摩尔线程GPU,打造通用计算MUSA核心软件栈底座,构建MUSA应用生态基石;为AI训练与推理、科学计算、物理仿真等应用提供极致优化的核心算子库;跨团队协作,深度参与硬件架构协同设计,推进软硬协同优化
岗位职责
- 基于摩尔线程GPU,打造通用计算MUSA核心软件栈底座,构建MUSA应用生态基石
- 为AI训练与推理、科学计算、物理仿真等应用提供极致优化的核心算子库
- 跨团队协作,深度参与硬件架构协同设计,推进软硬协同优化
- 开发并维护MUSA驱动、AI算子库、通信库、数学库等核心组件
- 面向用户、学生及广大开发者推广国产自主GPU通用计算软件生态
任职要求
- 计算机、自动化、人工智能、数学、物理、高性能计算等相关专业本科及以上学历,在校或应届毕业生优先
- 对C/C++编程及算法设计优化有浓厚兴趣
- 对底层软硬件架构有强烈兴趣,具备优雅的技术审美和极致的性能追求
- 积极乐观,责任心强,工作认真细致,具备良好的服务意识及优秀的团队沟通与协作能力
- 了解某一种异构加速芯片架构及其并行编程模型(如GPU、AI芯片等)者优先
- 熟悉CUDA/OpenCL/HIP/OneAPI/Vulkan等异构通用计算编程语言者优先
- 有高性能计算、异构计算相关知识积累或项目/实习经验者优先
- 在国际顶尖学术会议或期刊发表过论文者优先
- 在ASC、ACM、NOI等计算机科学相关竞赛中获奖者优先
- 了解机器学习、深度学习基础理论,具备优秀数学功底及算法设计能力者优先
- 具备优秀的软件工程能力、学习能力及自我驱动力
- 具备良好的学术论文阅读理解能力及文档撰写习惯
加分项
- 了解某一种异构加速芯片架构及其并行编程模型:如GPU、AI芯片等
- 了解某一种异构通用计算编程语言:CUDA/OpenCL/HIP/OneAPI/Vulkan等
- 在某一种异构平台上进行过并行算法的开发调优,有高性能计算、异构计算相关知识积累、项目/实习经验
- 在国际顶尖会议或期刊(CVPR、ICCV、ECCV、3DV、SIGGRAPH、TOG、TPAMI、ACL、NAACL、EMNLP、AAAI等)上发表过学术论文
- 在有影响的计算机科学相关的竞赛中获奖,如ASC、ACM、NOI等
- 了解机器学习、深度学习基础理论知识,对AI相关的工作有强烈的兴趣
- 优秀的数学功底,算法理解和设计能力
- 学习能力过硬,有强烈的自我驱动力
- 优秀的学术论文阅读理解能力,良好的文档撰写能力和习惯
- 丰女士 刚刚活跃
- 摩尔线程 · 人事HRBP
福利待遇
- 薪资范围:20,000–40,000元/月,共16薪
- 工作地点可选:北京、上海、杭州、成都
- 加入国产GPU领军企业,参与核心技术研发,职业发展空间广阔
工作地址
北京朝阳区望京国际研发园I座