AI算子开发工程师

AI算子开发工程师

工作地点：上海/杭州

岗位职责

参与 AI 芯片算子库的设计、开发与性能优化，支持深度学习模型在芯片上的高效运行；

负责矩阵计算、卷积、Attention 等核心算子的实现与优化，持续提升计算效率和硬件利用率；

针对不同模型和硬件架构进行性能分析、瓶颈定位及算子调优；

参与算子与深度学习框架、编译器及 Runtime 的适配和集成；

与编译器及框架团队协作，推动软硬件协同优化。

任职要求

本科及以上学历，计算机、软件工程、电子信息、数学等相关专业；

具备扎实的数据结构、算法及计算机体系结构基础，熟悉 C / C++ 或 Python；

了解并行计算及 GPU / NPU 等加速器基本原理，有 CUDA / Triton 等并行编程经验者优先；

熟悉 PyTorch 等深度学习框架，了解常见神经网络模型及基础算子；

了解 GEMM、Conv、Attention 等常见算子，有性能分析或优化相关经验者优先；

具备较强的逻辑分析、问题定位和学习能力，对性能优化有钻研精神和技术热情。

加分项

有 Triton Kernel、CUTLASS、FlashAttention、TVM / MLIR 等相关项目、科研、实习或竞赛经历者优先。