独角兽

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

壁仞科技

北京 / 上海 / 其他人工智能社会招聘

岗位描述

针对自研架构设计和优化Kernel综合器,支持硬件架构验证及高性能算子交付;负责核心基础设施的维护与演进,包括多层IR及Lower过程、编程范式抽象、面向Tensor抽象、ISA层抽象;负责关键Pass的设计与优化,如生命周期/活跃性分析、资源分配、Peephole优化;建设与维护回归体系,包括单元测试、集成测试、CI稳定性、代码规范与静态检查;与硬件团队协同设计,反馈自研指令集的性能与易用性问题;与算子、性能工具、硬件设计团队协作,沉淀设计文档与开发指南。要求精通C++(C++14及以上),具备扎实的软件工程能力,熟悉Linux开发环境与构建测试体系,理解并行体系结构与性能基础,具备良好的沟通协作能力和学习能力。有编译原理、LLVM/MLIR/TVM/Triton/TileLang、汇编级性能优化、DSL设计、性能建模等经验者优先。