算子库架构/工程师
岗位摘要
参与AI芯片算子库（CNN, Transformer, GNN等）的整体架构设计与演进；负责核心算子（卷积、矩阵乘、激活、归一化、注意力机制等）的高效实现与深度优化；研究前沿AI算法及其计算特点，将新算法高效映射到硬件平台
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口，并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 参与AI芯片算子库（CNN, Transformer, GNN等）的整体架构设计与演进
- 负责核心算子（卷积、矩阵乘、激活、归一化、注意力机制等）的高效实现与深度优化
- 研究前沿AI算法及其计算特点，将新算法高效映射到硬件平台
- 与编译器团队、驱动团队、AI框架团队紧密协作，优化算子调用链
- 进行算子性能分析与调优，定位性能瓶颈，持续优化算子库效率和质量
任职要求
- 计算机、人工智能、数学、电子工程等相关专业硕士及以上学历，优秀本科生亦可
- 深入理解深度学习基本原理和常用模型（CNN, RNN, Transformer等）
- 熟悉计算机体系结构，了解CPU/GPU/AI加速器的并行计算架构和内存层次结构
- 精通C/C++，具备扎实的编程功底和良好的代码风格
- 熟悉CUDA、OpenCL、HIP、OpenMP、SYCL等至少一种并行编程模型
- 具备较强的性能优化意识和能力，有底层代码优化经验者优先
- 了解AI框架（PyTorch, TensorFlow）基本原理和使用者优先
加分项
- 有GPU/AI芯片算子开发、优化或相关项目经验
- 熟悉深度学习算子融合、量化、稀疏化等优化技术
- 有LLVM、MLIR等编译器基础设施使用或开发经验
- 在AI系统、高性能计算相关会议（如PPoPP, SC, ICS, MLSys等）发表过论文者优先
- 有数学建模竞赛、编程竞赛等获奖经历
- 你将获得： 深入AI芯片软件栈最核心环节，亲手打造驱动AI应用落地的“发动机”
- 直接影响芯片的最终性能表现，挑战极致优化的技术高峰
- 与算法、硬件、框架等多领域专家协作，成为复合型AI系统人才
- 有竞争力的薪酬包、完善的福利体系和清晰的职业发展路径
- 李先生 刚刚活跃
- 壁仞科技 · 高级工程师
工作地址
杭州滨江区海威天地T3-28
登录后查看完整岗位详情
使用邮箱验证码登录后，可查看完整职责、任职要求、原岗位入口，并继续使用收藏和浏览记录。列表摘要保持公开，完整详情用于保护数据质量和降低恶意抓取。