岗位描述(节选)
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算子开发与优化经验……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。
本站当前在招 3324 个北京技术研发岗位,其中 86 个公开标注了薪资。按这些标注统计:月薪(税前)主流区间 12K ~ 28K(25~75 分位),中位约 20K,最低 2K、最高 60K。
在招岗位较多的企业:经纬恒润 590 个美团 313 个理想汽车 248 个字节跳动 240 个阿里巴巴集团 226 个
统计口径:求职者营地实时在招岗位的公开薪资标注,未标注者不计入;区间为 25~75 分位,非薪酬承诺。