阿里巴巴
AI推理平台-AI 高性能计算资深专家-杭州/北京/上海
官网发布时间：
职位描述
负责 AI 高性能计算的技术规划与核心问题攻坚，深入探索 NVIDIA、AMD、华为昇腾等 AI 芯片的底层架构，通过原生 Kernel 开发、Triton/TileLang 编译优化等手段，提升模型训练与推理性能。主导关键算子与算子库建设，推动优化成果在实际业务中落地，并指导团队提升底层开发与性能优化能力。
任职要求
1. 计算机、电子工程或相关专业本科及以上学历，对计算机体系结构和 GPU/NPU 的执行机制、存储层级有深入理解。 2. 拥有深厚的 AI 高性能计算经验，精通至少一种异构计算平台及编程模型（如 CUDA、ROCm、CANN 等），能够独立解决复杂的底层性能问题。 3. 深入理解 Transformer 等现代 AI 模型的计算特点，在 GEMM、Attention、MoE、低精度计算等方向有突出的优化经验和代表性成果。 4. 具备从零设计、实现高性能 Kernel 的能力，能够结合硬件特性开展优化，并将算子性能收益转化为实际训练或推理的效率提升。 5. 曾主导核心算子库、计算优化框架或重大性能优化项目，能够制定技术路线、判断方案取舍，并推动跨团队协作与落地。 6. 具备技术指导和人才培养能力，能够带领团队攻克关键问题，建立有效的开发与性能优化方法。 加分项： - 熟悉 Triton、TileLang、CUTLASS、TVM 等工具，或具备 MLIR/LLVM 相关开发经验。 - 有跨芯片平台优化、大规模训练与推理优化经验。 - 在相关开源项目中有重要贡献，或有具有行业影响力的技术成