腾讯
AI计算库开发工程师(北京/深圳)
AI计算库开发工程师(北京/深圳)
发布于 2 小时前普通员工/个人贡献者
全职员工仅现场办公高性能计算
高性能计算
Pytorch
Tensorflow
并行编程
异构计算
性能调优
算子优化
高性能计算
值不值得投
75综合评分
综合评分 75
自研AI芯片核心研发岗，技术前沿且成长空间巨大，薪资潜力高，但底层优化挑战大且WLB情况未知。
从薪资福利、成长发展、工作生活和使命价值四个维度综合评估，方便横向比较。
薪资怎么样
5万
比高性能计算中位数高
发布情况
新岗位 · 今天发布
公司情况
这家公司招聘很活跃
约 1960 个在招
市场机会
选择不多
深圳 · 高性能计算 · 4 个在招
职位详情
关于这个职位
该职位主要负责基于自研AI芯片进行高性能算子的研发与优化，设计高效的数据流与计算流方案
你需要与编译器、runtime等团队紧密协作，推动深度学习模型在自研芯片上的端到端性能优化与落地
最低要求
熟悉 Linux 开发环境，扎实掌握 C++/Python，具备良好的代码能力和工程能力
熟悉异构计算或高性能计算基础，理解并行编程、向量化、线程/任务调度、访存优化等基本原理
熟悉 PyTorch、TensorFlow 等主流深度学习框架
具备性能瓶颈定位与调优能力，具备profile和debug工具使用经验
工作职责
基于自研 AI 芯片软件栈以及架构特点，研发和优化高性能算子，设计高效的数据流与计算流方案
结合芯片计算单元和存储层次，构建可维护、可扩展的算子实现
参与测试框架和性能分析体系的建设，提升算子研发效率、质量保障能力和生态易用性
与编译器、runtime及业务团队协作，推动模型在自研芯片上的端到端性能优化与落地
AI 洞察
优缺点分析
优点
- 参与自研AI芯片核心底层研发，技术壁垒高，能积累深厚的异构计算与算子优化经验
- 平台资源顶级，能接触到最前沿的AI硬件生态与大规模业务落地场景
- 底层算子优化与芯片架构结合难度大，需要持续跟进快速迭代的AI模型与硬件架构
- 适合对底层系统、芯片架构和高性能计算有浓厚兴趣，具备扎实C++功底和极客精神的研发工程师
缺点 / 挑战
- 跨团队协作（编译器、runtime等）沟通成本较高，对端到端性能负责的压力较大
角色解读
- 可向AI芯片底层架构专家、高性能计算技术专家方向纵深发展
- 也可转型为AI基础设施团队的技术负责人，主导自研芯片生态建设与业务赋能
- 负责自研AI芯片的高性能算子研发与优化，设计数据流与计算流方案
- 结合芯片架构构建可扩展的算子实现，并参与测试与性能分析体系建设
- 跨团队协作推动深度学习模型在自研芯片上的端到端性能优化与落地
- 精通C++/Python及Linux开发环境，具备优秀的工程实现能力
- 深入理解异构计算、并行编程及访存优化等高性能计算底层原理
- 熟悉PyTorch/TensorFlow等主流框架，具备性能瓶颈定位与调优经验
申请策略
- 了解自研AI芯片的最新进展与生态布局，面试时展现出对国产算力生态发展的热情与思考
- 突出在异构计算、CUDA/ROCm或自研芯片上的算子开发与优化经验
- 展示使用profile工具进行性能瓶颈定位并显著提升计算效率的具体量化案例
- 强调对PyTorch/TensorFlow底层调度机制的理解及自定义算子接入经验
- 深入学习计算机体系结构与芯片存储层次原理，强化访存优化能力
- 熟悉主流AI编译器（如TVM、MLIR）及Runtime机制，拓宽技术视野
面试指南
- 采用STAR法则，先描述业务背景与性能目标，再详细说明使用的profile工具和定位过程，最后给出具体的优化手段（如向量化、内存合并、流水线并行）及量化收益
- 请详细描述一个你做过的高性能算子优化案例，你是如何定位瓶颈并提升性能的？
- 在异构计算中，如何平衡计算单元与存储层次的数据搬运开销？
- PyTorch/TensorFlow的算子调度机制是怎样的？如何自定义算子并接入框架？
- 面对一个复杂的深度学习模型，你如何设计端到端的性能分析与优化方案？
- 复习计算机体系结构、并行计算原理及CUDA编程，夯实底层理论基础
- 准备1-2个深度的性能优化项目案例，能够讲透底层原理与调优细节
职位点评
薪资福利75
成长发展90
工作生活50
使命价值85
薪资福利
75中等
JD未披露具体薪资数字与福利条款，无法直接评估其补偿性动机满足程度，需结合面试沟通确认。
薪资信号未披露（AI估算：3.5千–6.5万/月）
成长发展
90较高
岗位涉及自研AI芯片底层算子与高性能计算，技术壁垒极高，处于AI硬件生态前沿，个人技术成长空间巨大。
技术前沿前沿/新兴技术
技术栈C++、Python、异构计算、高性能计算、PyTorch、TensorFlow、算子优化
业务类型ambiguous
工作生活
50较低
JD未提及任何工作模式、办公灵活性或WLB相关信息，无法判断其实际工作节奏与生活平衡情况。
工作模式未明确
办公地点科技园/产业园
加班情况未提及（无法判断）
使命价值
85较高
参与自研AI芯片研发，助力国产算力生态建设，具有极高的行业价值与技术使命感。
行业发展高速增长赛道
社会影响正向社会影响力较高
创新程度开拓性创新（行业首创）
Watch Jobs