AI推理平台-AI 高性能计算工程师-杭州/北京/上海

关于这个机会

AI 高性能计算工程师负责探索不同AI芯片（NV，AMD, 华为昇腾, TPU, 寒武纪等）的底层架构，使用硬件手写原生 Kernel、 Trition/Tilelang 编译优化等手段，解决“从0到1”（跑通）和“从1到N”（跑得快）的关键问题。

任职要求

1. 计算机、电子工程或相关专业本科及以上学历，对计算机体系结构有深刻理解。 2. 拥有深厚的GPU/NPU/XPU高性能计算优化经验，精通至少一种异构计算平台及编程模型（如CUDA, ROCm, OpenCL, SYCL, CANN等）。 3. 熟悉现代AI模型（特别是Transformer架构）中关键算子的计算原理和优化方法。 具备以下至少一项经验者优先： -有从零手写高性能算子（Kernel）并取得显著性能收益的成功案例。 -熟悉编译器技术，有MLIR/LLVM相关开发经验，或有丰富的Triton/TVM算子开发与优化经验。 -有针对AMD、华为、Intel等特定AI芯片的深度性能调优经验。 -具备极强的技术钻研精神和解决复杂问题的能力，能够适应快速迭代的技术环境。

内推申请说明

本站为独立内推协助平台。申请会交由管理员核实岗位与内推渠道，不等于已在公司官网投递；薪资、岗位状态和实际招聘流程以官方信息为准。