岗位描述
AI 编译器开发工程师
岗位职责
1. 负责公司自研 AI 芯片编译器后端开发,包括基于 LLVM 的指令选择(Instruction Selection)、代码生成(Code Generation)等编译器后端能力建设。
2. 利用 MLIR、TVM 等 AI 编译器技术,支持深度学习模型及算子在公司自研 AI 芯片上的编译、Lowering(逐级降级转换)及性能优化。
3. 负责 AI 编译器关键模块的设计与开发,包括:中间表示(IR)、编译器 Pass、算子融合(Operator Fusion)、计算图优化(Graph Optimization)、代码生成(Code Generation)。
4. 参与面向公司自研 AI 芯片的 C/C++ 语言扩展开发,包括 Clang/LLVM 编译器支持、IR 转换、后端代码生成适配及性能优化。
5. 与算法、算子/Kernel、深度学习框架团队协作,推动 大语言模型(LLM)及主流 AI 模型在公司自研 AI 芯片上的高效部署,解决编译器相关问题及跨软件栈性能瓶颈。
6. 参与编译器整体架构设计、性能分析及复杂技术问题排查,持续提升编译质量、生成代码性能及软件栈稳定性。
任职要求
1. 计算机科学、软件工程、电子工程或相关专业,本科及以上学历。
2. 熟练掌握 C/C++,具备扎实的数据结构、算法、操作系统及计算机体系结构基础。
加分项
1. 具有 LLVM Backend、编译器后端开发或面向新硬件架构的 CodeGen 开发经验。
2. 具有 GPU、NPU、TPU 等异构计算平台的编译器开发经验。
3. 熟悉 CUDA、OpenCL、HIP 等异构计算编程模型。
4. 有阅读、研究、参与贡献或实际使用大型开源项目的经验,包括:PyTorch、TensorFlow、Triton、TVM、MLIR。
5. 具有 AI 模型、算子/Kernel 开发或深度学习框架相关经验。
6. 具有编译器或 AI 编译器相关开源贡献经验。
7. 具有以下经验者优先:Clang Frontend 开发、编程语言扩展、自定义 Builtins / Intrinsics、面向新硬件架构的编程语言扩展设计与开发。