AI 编译器工程师(端到端编译栈方向)
北京微电子技术研究所岗位职责:1、参与端到端 AI 编译栈(TVM / IREE / torch.compile+Inductor / XLA)的建设与定制:从 PyTorch / ONNX 计算图导入,到图优化、算子融合、代码生成与部署; 2、参与 MLIR 侧 Dialect 与 Pass 开发:图级变换(算子融合、layout 与 dtype 传播、常量折叠)、Bufferization、lowering 到硬件 Dialect; 3、参与自动调度与调优:调度空间搜索(AutoTVM / MetaSchedule)、tiling 与循环变… 岗位要求:基本要求 1、计算机 / 软件工程 / 人工智能等相关专业; 2、熟悉 C++ 与 Python,具备大型工程代码的阅读与调试能力; 3、了解编译原理基础(中间表示、Pass、数据流分析)与深度学习模型基本结构(Transformer 与常见算子); 4、熟悉 PyTorch,能跑通模型并理解其计算图与算子调用链路。 加分项 1、用过或改过 TVM / IR…