AI编译优化工程师(北京/上海/深圳) 上海 分享

分享岗位

方式1:复制岗位链接

方式2:分享岗位海报

手机扫描二维码分享

收藏 TEG 点击了解更多BG信息 技术 三年以上工作经验 更新于年06月25日 岗位职责

1.基于自研芯片，研发高性能图和算子编译器，解决芯片落地过程中编译相关性能、易用性等问题；
2.探索和研发自动生成算子解决方案，提升生成算子效率；
3.不断迭代编译器功能、性能和易用性，和业务一起构建AI生态。

岗位要求

1.熟悉Linux开发环境，掌握Python/C++等语言， 有良好的编程基础、系统设计优化能力；
2.熟悉GPU/AI DSA体系结构和原理，有Nvidia/AMD/Intel或 AI 芯片等编译相关的开发优化经验；
3.精通MLIR、Triton、tilelang或其他CodeGen框架；
4.具备良好的技术热情和责任感，优秀的分析解决问题能力，良好的团队意识和沟通能力。

加分项

1.熟悉Cuda、ROCm算子开发与pytorch/tensorflow框架；
2.熟悉LLM常见模型结构和大模型相关的模型并行、流水并行方法。