深度学习高性能计算研发工程师
岗位摘要
负责AI模型在摩尔线程芯片的适配与性能优化,协助客户完成业务落地;与客户技术团队并肩工作,解决使用摩尔技术栈时的复杂技术挑战;提供GPU架构、AI算法、软件架构优化方案,实现极致GPU加速
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 负责AI模型在摩尔线程芯片的适配与性能优化,协助客户完成业务落地
- 与客户技术团队并肩工作,解决使用摩尔技术栈时的复杂技术挑战
- 提供GPU架构、AI算法、软件架构优化方案,实现极致GPU加速
- 跟踪大模型推理、训练、编译器优化及硬件加速领域最新技术动态,持续改进软硬件产品
任职要求
- 计算机科学相关专业硕士或博士,3年以上相关经验
- 深厚的C++和Python编程能力,具备良好编程风格与系统设计能力
- 熟悉深度学习框架原理,如Megatron/DeepSpeed/ColossalAI/torchtitan/sglang/vllm,有大模型推理训练开发及深度优化经验优先
- 具备LLM、AIGC、AI Agent、3D等算法经验优先
- 有AI编译器(XLA、TVM、MLIR)开发经验优先
- 熟悉GPU架构及底层硬件原理,具备CUDA并行编程与算子开发经验优先
- 出色的分析与解决问题能力,热衷挑战高难度技术难题
- 具备优秀公开演讲与技术分享能力
- 沟通表达优秀,具备团队合作精神,能自我驱动,适应快节奏环境
加分项
- 在LLM,AIGC,AI Agent,3D等方向有相关算法经验优先
- 有AI编译器(如XLA、TVM、MLIR)等软件栈的开发经验优先
- 熟悉GPU架构以及底层硬件原理,具备CUDA并行编程和算子开发经验优先
- 出色的分析和解决问题的能力,热衷于解决极具挑战性的技术难题
- 具备出色的公开演讲和技术分享能力
- 优秀的沟通和表达能力,具备团队合作精神,能够自我驱动,并在快节奏的环境中茁壮成长
- 王女士 刚刚活跃
- 摩尔线程 · 部门负责人
福利待遇
- 与顶尖客户并肩解决前沿技术难题,技术成长迅速
- 深度参与国产GPU生态建设,影响下一代AI加速产品定义
- 开放技术分享氛围,鼓励公开演讲与内部技术交流
- 望京核心地段办公,交通便利,团队氛围高效融洽
工作地址
北京朝阳区望京国际研发园I座
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。