壁仞科技
2027届民企

岗位信息

招聘岗位
2027校招-算子库架构/工程师
工作城市
上海、北京、杭州
是否笔试
未明确
招聘批次
2027届
发布日期
2026-08-14

📋 岗位职责 / 任职要求

岗位职责: 1.参与AI芯片算子库(如CNN, Transformer, GNN等算子)的整体架构设计与演进,确保其高性能、高能效、易用性和可扩展性。 2.负责核心算子(如卷积、矩阵乘、激活、归一化、注意力机制等)的高效实现与深度优化,充分利用底层硬件特性(如SIMD/SIMT、共享内存、缓存、特定加速单元等)。 3.研究前沿AI算法及其计算特点,将新算法高效映射到硬件平台,设计并实现对应的高性能算子。 4.与编译器团队、驱动团队、AI框架团队(如TensorFlow, PyTorch,MindSpore等)紧密协作,定义算子接口,优化算子调用链,提升端到端性能。 5.进行算子性能分析与调优,定位性能瓶颈,持续优化算子库的效率和质量。 6.参与算子库相关文档的编写,为内外部用户提供技术支持。 任职要求: 学历背景: 计算机、软件工程、人工智能、数学、电子工程等相关专业 专业基础: 1.深入理解深度学习基本原理和常用模型(CNN, RNN, Transformer等)。 2.熟悉计算机体系结构,了解CPU/GPU/AI加速器的并行计算架构和内存层次结构。 3.扎实的数据结构与算法基础。 技能要求: 1.精通C/C++,具备扎实的编程功底和良好的代码风格。 2.熟悉至少一种并行编程模型和语言,如CUDA, OpenCL, HIP,OpenMP, SYCL等。 3.具备较强的性能优化意识和能力,有底层代码优化经验者优先。 4.了解AI框架(如PyTorch, TensorFlow)的基本原理和使用者优先。 5.具备良好的问题分析与解决能力,以及较强的学习能力和团队协作精神。 加分项: 1.有GPU/AI芯片算子开发、优化或相关项目经验。 2.熟悉深度学习算子融合、量化、稀疏化等优化技术。 3.有LLVM、MLIR等编译器基础设施使用或开发经验。 4.在AI系统、高性能计算相关会议(如PPoPP, SC, ICS, MLSys等)发表过论文者优先。 5.有数学建模竞赛、编程竞赛等获奖经历。

🏢 企业简介

总部上海

投递方式

投递入口为会员专属,登录 / 注册 后查看(注册送 15 天免费试用)

ℹ️ 信息来源

本条招聘信息由校招宝转载自 Moka 招聘系统(企业校招官网),版权归原发布方所有,校招宝仅作信息聚合与导航。
招聘信息以官方原文为准,投递前请核实截止日期与要求。如涉侵权或信息有误,请邮件联系 kknee@qq.com,我们将在 24 小时内处理。