返回岗位列表

壁仞科技

AI 编译器架构工程师(kernel 生成方向)

地点:北京 薪资:40-60K 日期:2026-03-03

岗位摘要

针对自研架构,设计和优化 kernel 综合器,支持硬件架构验证及高性能算子交付;负责核心基础设施的维护与演进,包括多层 IR 及 lower 过程、编程范式抽象等;负责关键 Pass 的设计与优化,如生命周期分析、资源分配、Peephole 优化

岗位职责

  • 针对自研架构,设计和优化 kernel 综合器,支持硬件架构验证及高性能算子交付
  • 负责核心基础设施的维护与演进,包括多层 IR 及 lower 过程、编程范式抽象等
  • 负责关键 Pass 的设计与优化,如生命周期分析、资源分配、Peephole 优化
  • 建设与维护回归体系,包括单元测试、集成测试、CI 稳定性及代码规范检查
  • 与硬件团队协同设计,反馈自研指令集的性能与易用性问题
  • 与算子、性能工具及硬件设计团队协作,沉淀设计文档与开发指南

任职要求

  • 精通 C++(C++14 或以上),具备扎实的软件工程能力,如设计模式、模块化设计
  • 熟悉 Linux 开发环境与构建测试体系,包括 CMake、单元测试框架及 CI 实践
  • 理解并行体系结构与性能基础,如 SIMT/GPU 模型、内存层级、同步与一致性概念
  • 具有良好的沟通协作能力,能输出清晰的技术方案与文档
  • 具有好奇心与学习能力,并在之前的项目中得到验证
  • 具备编译原理经验,如 IR/CFG、数据流分析、寄存器分配或指令选择调度者优先
  • 有 LLVM/MLIR/TVM 等编译器或 kernel codegen 相关经验者优先

加分项

  • 有编译原理经验:IR/CFG、数据流分析、live interval、寄存器分配、指令选择/调度、DCE/拷贝传播等
  • 有LLVM/MLIR/TVM/Triton/TileLang等编译器或kernel codegen相关经验
  • 有汇编级性能优化、资源分配、指令重排、手排任务级流水等经验
  • 有基于公式的理论性能建模,性能计数器设计与使用经验
  • 有vibe coding经验:复杂项目及多人协作项目的构建及管理
  • 马先生 刚刚活跃

工作地址

北京朝阳区浦项中心B座10F