返回岗位列表

2026校招-算子库架构/工程师

查看原岗位

地点：北京薪资：20-40K 日期：2026-03-03

岗位摘要

参与AI芯片算子库整体架构设计与演进，确保高性能、高能效、易用性和可扩展性；负责卷积、矩阵乘、激活、注意力等核心算子高效实现，深度挖掘硬件特性；研究前沿AI算法，设计并映射高性能算子至硬件平台

岗位职责

参与AI芯片算子库整体架构设计与演进，确保高性能、高能效、易用性和可扩展性
负责卷积、矩阵乘、激活、注意力等核心算子高效实现，深度挖掘硬件特性
研究前沿AI算法，设计并映射高性能算子至硬件平台
与编译器、驱动、AI框架团队协同定义接口并优化端到端性能
进行性能分析与调优，定位瓶颈，持续提升算子库质量
编写技术文档，为内外部用户提供支持

任职要求

计算机/人工智能/电子工程等相关专业硕士及以上，优秀本科亦可
深入理解CNN/RNN/Transformer等深度学习原理与模型
熟悉CPU/GPU/AI加速器并行架构及内存层次结构
精通C/C++，掌握CUDA/OpenCL/HIP等至少一种并行编程模型
具备底层性能优化经验，了解PyTorch/TensorFlow框架原理
具备优秀的问题分析、学习能力与团队协作精神
有GPU/AI算子开发、LLVM/MLIR经验或顶会论文者优先
数学建模/编程竞赛获奖经历加分

加分项

有GPU/AI芯片算子开发、优化或相关项目经验
熟悉深度学习算子融合、量化、稀疏化等优化技术
有LLVM、MLIR等编译器基础设施使用或开发经验
在AI系统、高性能计算相关会议（如PPoPP, SC, ICS, MLSys等）发表过论文者优先
有数学建模竞赛、编程竞赛等获奖经历
你将获得：深入AI芯片软件栈最核心环节，亲手打造驱动AI应用落地的“发动机”
直接影响芯片的最终性能表现，挑战极致优化的技术高峰
与算法、硬件、框架等多领域专家协作，成为复合型AI系统人才
有竞争力的薪酬包、完善的福利体系和清晰的职业发展路径
田先生刚刚活跃
壁仞科技 · 芯片架构工程师

福利待遇

深入AI芯片软件栈核心，亲手打造AI应用“发动机”
直接影响芯片最终性能，挑战极致优化技术高峰
与算法、硬件、框架专家协作，成长为复合型AI系统人才
提供有竞争力薪酬15薪、完善福利及清晰职业发展路径

工作地址

上海闵行区浦江科技广场16号楼13楼