飞腾信息技术有限公司
Triton/AI编译器开发工程师
岗位职责
Triton 编译器开发:基于自研 GPGPU 硬件指令集及特性,搭建基于 MLIR 框架的 AI 编译器,负责异步调度、向量化、软件流水线、内存规划、后端代码生成等核心特性开发与调优,支撑 Triton Kernel 的正确高效运行。
框架生态接入:接入与扩展 TorchInductor、vLLM 等框架的 Triton 编译运行过程,支撑主流 LLM 在自研硬件上的端到端运行。
跨团队协作:承接上层应用框架团队下发的各种 Kernel 类型,完成优化调优;与后端编译器、驱动团队对齐底层运行行为,合作解决相关问题。
任职要求
教育背景:计算机、应用数学、电子信息、通信工程等相关专业硕士及以上学历。
底层理论储备: - AI 编译器知识:熟悉编译器架构与 IR 逐级 Lower 设计思想,掌握 SSA、控制流、Data Layout 等基本抽象; - 并行编程知识:熟悉 GPGPU 并行编程基础概念,了解内存层次、线程分层结构、同步与异步指令等; - LLM 算法知识:熟悉 FlashAttention、MoE 等 LLM 关键算法原理与实现。
技术栈要求: - 精通现代 C/C++/Python 编程,具备良好的抽象建模能力及工程实践习惯,熟练掌握 Git/CMake/LLDB/PyTest 等相关工具的使用; - 熟悉 MLIR 编译器基础设施,具备实际开发经验,能设计并实现 Dialect、Pass、Translation 等。 薪资: 面议 学历要求(及以上): 本科 工作经验: 5-8年 年龄要求: 不限 其他要求:
加分项
有异构并行编程和调试经验,了解 CUDA/HIP/CANN/OpenCL/SYCL 等框架;
有 Triton/CuTile/TileLang 等并行编程框架的编译器、运行时及算子开发经验;
有实际的 LLM 部署加速经验,熟悉 vLLM、SGLang 等 LLM 推理框架;
熟悉常见 GPGPU 或 NPU 的硬件架构与指令执行细节(如内存系统、SIMD 单元、SIMT、TensorCore 单元、DMA 单元等)。
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。