阿里巴巴控股集团 · 社会招聘
AI推理平台-AI 高性能计算专家-大模型推理
杭州 ·3年以上·本科
北京·杭州
薪资面议
前往 阿里巴巴控股集团 官网投递
职位描述
我们是阿里巴巴大模型推理团队,负责内部 LLM/AIGC 百炼推理服务建设,为淘宝、天猫、聚划算、优酷、闲鱼等多个集团业务部门提供强有力的技术支撑和底层服务能力。
AI 高性能计算工程师负责探索不同AI芯片(NV,AMD, 华为昇腾, TPU, 寒武纪等)的底层架构,使用硬件手写原生 Kernel、 Trition/Tilelang 编译优化等手段,解决“从0到1”(跑通)和“从1到N”(跑得快)的关键问题。
工作地点:北京、杭州,未来会开放「上海」工作地点,现阶段支持员工根据团队安排部分时间在上海办公,详情可与hr或面试官沟通。
【任职要求】
1. 计算机、电子工程或相关专业本科及以上学历,对计算机体系结构有深刻理解。
2. 拥有深厚的GPU/NPU/XPU高性能计算优化经验,精通至少一种异构计算平台及编程模型(如CUDA, ROCm, OpenCL, SYCL, CANN等)。
3. 熟悉现代AI模型(特别是Transformer架构)中关键算子的计算原理和优化方法。
具备以下至少一项经验者优先:
1. 有从零手写高性能算子(Kernel)并取得显著性能收益的成功案例。
2. 熟悉编译器技术,有MLIR/LLVM相关开发经验,或有丰富的Triton/TVM算子开发与优化经验。
3. 有针对AMD、华为、Intel等特定AI芯片的深度性能调优经验。
4. 具备极强的技术钻研精神和解决复杂问题的能力,能够适应快速迭代的技术环境。
最后确认在招:2026-09-01 19:10:48 · 来源平台:alibaba