资深大模型AI编译器工程师

该职位来源于猎聘

岗位职责：

开发将AI编译器以及算子库，一些工作职责包括：

• 主导 AI 编译器工具链的设计与优化，深入参与编译流程、IR 构建、核心 Pass 的研发； 2. 负责模型与算子的导入、转化、融合、拆分等模块的方案及落地实现； 3. 深度理解芯片指令集与硬件架构，结合硬件特性制定算子调度与执行加速策略； 4. 基于Triton / MLIR / IREE等开源框架开发自定义算子与编译优化组件； 5. 参与运行时算子库的实现与性能优化，打通从编译到执行的全链路； 6. 与模型、框架、硬件团队紧密协作，打造适配大模型推理的高性能编译系统。

任职要求：

• 计算机科学、电子工程等相关专业硕士及以上学历； 2.5年以上编译器开发经验，其中至少2年以上AI编译器开发经验； 这样也行。条件宽松一些；； 3. 深入理解 AI 编译流程（前端 / 中端 / 后端）之一，并具备实际落地经验； 4. 熟悉大语言模型并行拆分与算子加速，理解 Attention等核心模块的性能瓶颈与优化手段 5. 熟悉至少一种主流编译框架（TVM / XLA / MLIR），具备相关开发经验； 6. 熟悉 PyTorch 源码与编译机制（torch.compile / torch.\_dynamo / torch.fx 等）优先； 7. 理解硬件架构（CPU / GPU / NPU / 自研芯片）与低层性能优化技术（向量化、流水线、内存调度等）； 8. 具备良好的沟通表达能力与技术文档能力，能够将复杂方案清晰化、可落地化。