壁仞科技
2027校招-算子库架构/工程师
岗位职责
参与AI芯片算子库（如CNN, Transformer, GNN等算子）的整体架构设计与演进，确保其高性能、高能效、易用性和可扩展性。
负责核心算子（如卷积、矩阵乘、激活、归一化、注意力机制等）的高效实现与深度优化，充分利用底层硬件特性（如SIMD/SIMT、共享内存、缓存、特定加速单元等）。
研究前沿AI算法及其计算特点，将新算法高效映射到硬件平台，设计并实现对应的高性能算子。
与编译器团队、驱动团队、AI框架团队（如TensorFlow, PyTorch,MindSpore等）紧密协作，定义算子接口，优化算子调用链，提升端到端性能。
进行算子性能分析与调优，定位性能瓶颈，持续优化算子库的效率和质量。
参与算子库相关文档的编写，为内外部用户提供技术支持
任职要求
学历背景： 计算机、软件工程、人工智能、数学、电子工程等相关专业 专业基础：
深入理解深度学习基本原理和常用模型（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等）发表过论文者优先。
有数学建模竞赛、编程竞赛等获奖经历
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理，最终申请以企业官方页面为准。