北极雄芯信息科技(西安)有限公司
AI 编译器(云计算芯片)工程师(西安)
RESPONSIBILITIES
岗位职责
1、研究并优化 Triton 编译器,使其高效支持公司芯片架构;
2、基于 MLIR 框架优化编译流程,提升生成代码的执行效率;
3、参与图优化、后端代码生成、调度策略、指令优化及流水并行化等工作;
4、研究并优化 Kernel 计算性能,降低计算开销,提高系统吞吐率;
5、研究算子融合(Op Fusion)、自动调优(Auto-Tuning)和代码生成(CodeGen)等优化策略;
6、与硬件团队、AI 框架团队协作,持续提升 AI 编译器的功能支持与性能表现;
7、跟踪 LLVM、Triton、MLIR 等 AI 编译器领域的前沿技术,持续优化编译性能。
REQUIREMENTS
任职要求
1、熟练使用 AI 工具,具备运用 AI 工具解决复杂疑难问题的经验;
2、电子、计算机等相关专业研究生及以上学历;
3、具备优秀的学习能力,对 AI 编译器和 AI 计算优化有浓厚兴趣;
4、熟悉 C++、Python,具备编译器或 AI 编译优化开发经验,熟悉 Triton、MLIR 编译流程;
5、熟悉 MLIR、Linalg 等编译基础设施;
6、具有 TVM、XLA 或自研编译器等 AI 编译优化经验者优先;
7、熟悉 Kernel 计算优化、算子融合、自动并行化等高性能计算优化策略者优先
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。
