岗位职责
AI 高性能计算工程师负责探索不同AI芯片(NV,AMD, 华为昇腾, TPU, 寒武纪等)的底层架构,使用硬件手写原生 Kernel、 Trition/Tilelang 编译优化等手段,解决“从0到1”(跑通)和“从1到N”(跑得快)的关键问题。
职位要求
- 计算机、电子工程或相关专业本科及以上学历,对计算机体系结构有深刻理解。
- 拥有深厚的GPU/NPU/XPU高性能计算优化经验,精通至少一种异构计算平台及编程模型(如CUDA, ROCm, OpenCL, SYCL, CANN等)。
- 熟悉现代AI模型(特别是Transformer架构)中关键算子的计算原理和优化方法。 具备以下至少一项经验者优先: -有从零手写高性能算子(Kernel)并取得显著性能收益的成功案例。 -熟悉编译器技术,有MLIR/LLVM相关开发经验,或有丰富的Triton/TVM算子开发与优化经验。 -有针对AMD、华为、Intel等特定AI芯片的深度性能调优经验。 -具备极强的技术钻研精神和解决复杂问题的能力,能够适应快速迭代的技术环境。