高性能计算工程师(训练/推理)
岗位摘要
负责AI模型在摩尔线程芯片的适配及性能优化,协助客户完成业务落地,协助内部团队完成产品定义与迭代;以客户为导向,与客户技术团队紧密合作,解决使用摩尔技术栈时遇到的复杂技术挑战;提供GPU架构、AI算法、软件架构等优化方案,实现极致的GPU加速
岗位职责
- 负责AI模型在摩尔线程芯片的适配及性能优化,协助客户完成业务落地,协助内部团队完成产品定义与迭代
- 以客户为导向,与客户技术团队紧密合作,解决使用摩尔技术栈时遇到的复杂技术挑战
- 提供GPU架构、AI算法、软件架构等优化方案,实现极致的GPU加速
- 跟踪大模型推理、训练、编译器优化及硬件加速等领域的最新技术动态,持续改进软件及硬件产品定义
任职要求
- 计算机科学相关专业,硕士及以上学位,2年以上相关工作经验,工程能力强
- 熟悉C++/Python,具有优秀的编程能力
- 熟悉深度学习框架工作原理,包括但不限于Megatron、DeepSpeed、Sglang、Vllm、VLM等,有大模型推理训练方向开发及深度优化经验者优先
- 熟悉GPU架构及底层硬件原理,具备CUDA并行编程和算子开发经验者优先
- 具备出色的分析和解决问题的能力,热衷于解决极具挑战性的技术难题
- 具有客户服务意识,拥有优秀的沟通表达能力和团队合作精神,能够自我驱动并在快节奏环境中高效工作
福利待遇
- 薪资范围:40,000–70,000元/月,共16薪
工作地址
北京朝阳区望京国际研发园l座3层