AgiBot logo
智元机器人
高性能算子开发实习生(R10506)

高性能算子开发实习生(R10506)

发布于 大约 1 小时前

实习/见习

上海市
无经验要求
实习生
仅现场办公
学历未注明
研究与开发 (研发)
Cuda
Gpu
Ncu
Nsys
Pytorch
Tensorflow
异构计算
算子开发

AI 估算 · 4k–8k

实习生岗位,上海AI公司,薪资水平对标市场中等水平。

职位详情

关于这个职位

该实习生岗位专注于高性能计算(HPC)领域的算子开发,你将参与CPU/GPU/ASIC多芯片的性能优化,使用CUDA、Triton、CUTLASS等工具提升AI模型的推理与训练效率

工作内容包括设计核心算法、解决计算性能瓶颈、参与异构计算集群构建,适合对并行计算和AI加速有浓厚兴趣的同学

最低要求

精通 CUDA 生态,熟悉编译流程,深入理解 GPU 架构、内存模型及并行计算原理,具备扎实的 CUDA 编程功底,能独立编写、调试并优化 CUDA Kernel

熟练使用 Triton、CUTLASS 等工具,有基于这些工具进行算子开发、模型加速的实战经验
理解 PyTorch/TensorFlow 等 AI 框架及常见 AI 模型(VLA, LLM等)
熟悉 Nsys, NCU 等性能分析工具,能精准定位并解决 GPU 计算中的性能瓶颈
熟悉国产芯片的运行流程及编译链路,具备相关算子开发实战经验

工作职责

负责 HPC 核心算法设计开发,优化 CPU/GPU/ASIC 多芯片性能,追求极致吞吐量与延迟

对接业务场景,定位并解决计算性能瓶颈,提升业务运行效率
参与异构计算集群的构建,推进异构计算方案落地与迭代

AI 洞察

优缺点分析

优点

  • 接触前沿HPC技术和AI加速领域,技能含金量高
  • 实习期间能深入学习CUDA生态和异构计算,提升底层能力
  • 公司处于机器人AI热门赛道,发展前景好
  • 技术要求高,需要扎实的CUDA和并行计算基础,学习曲线陡
  • 工作强度可能较大,涉及性能攻关和调优
  • 实习生需快速上手复杂工具和业务场景
  • 适合对高性能计算和AI加速有浓厚兴趣、具备较强编程和算法能力的计算机/电子相关专业学生

缺点 / 挑战

暂无明显挑战项

角色解读

  • 从算子开发实习生成长为高性能计算工程师,深入GPU底层优化
  • 可转向AI芯片编译器、异构计算架构师等方向
  • 积累经验后有机会参与核心AI训练推理框架的开发
  • 设计并实现高性能算子,使用CUDA、Triton等工具优化GPU/CPU计算性能
  • 分析业务场景中的计算瓶颈,定位性能问题并给出优化方案
  • 参与异构计算集群的构建与迭代,推动先进计算方案落地
  • 精通CUDA编程,熟悉GPU架构和内存模型,能独立编写Kernel
  • 熟练使用Triton、CUTLASS等算子开发工具,有实战经验
  • 理解常见AI框架(PyTorch/TensorFlow)和模型(LLM/VLA)
  • 熟悉性能分析工具(Nsys/NCU),善于性能调优

申请策略

  • 关注智元机器人的技术博客或开源项目,体现对公司的了解
  • 准备一个你主导的性能优化案例,展示量化分析能力
  • 突出CUDA编程相关项目经验,尤其是算子优化或Kernel开发
  • 展示使用Triton、CUTLASS等工具的实战案例,附性能提升数据
  • 强调对GPU架构的理解,可描述内存访问优化或并行计算技巧
  • 提前复习CUDA编程模型、内存层次结构、warp调度等
  • 动手实践Triton或CUTLASS,复现经典算子优化
  • 了解国产芯片(如华为昇腾)的编译流程和算子开发

面试指南

  • 对于原理性问题,先给出定义,再结合实际场景说明
  • 对于项目经验问题,采用STAR法则描述背景、任务、行动、结果
  • 对于性能优化问题,强调指标(延迟、吞吐)、工具(Nsys)和优化点(内存、计算)
  • 请解释GPU的warp调度机制以及如何影响性能?
  • 描述一次你使用CUDA进行算子优化的经历,使用了哪些技巧?
  • Triton与CUDA编程相比有哪些优劣势?
  • 如何分析一个AI模型在GPU上的计算瓶颈?
  • 了解哪些国产芯片?与NVIDIA GPU的差异是什么?

职位点评

67
综合评分

前沿HPC实习,技术成长极高,薪资较低,现场办公。

从学习成长、工作节奏、岗位方向和实习待遇综合评估,方便比较实习机会。

更适合这类人
适合追求技术成长和学习前沿HPC技能的求职者,对薪资和WLB要求不高。
表现最好
成长发展
相对薄弱
薪资福利
薪资福利40
成长发展90
工作生活50
使命价值70

薪资福利

40较低

实习生薪资相对全职较低,且无明确福利说明,补偿性动机满足有限。

薪资信号未披露(AI估算:4K-8K/月)

成长发展

90较高

该岗位提供前沿HPC技术学习和实战机会,技能成长空间大,发展性动机满足度高。

技术前沿前沿/新兴技术
技术栈CUDA、Triton、CUTLASS、异构计算、GPU优化
业务类型profit_center

工作生活

50较低

实习岗位通常要求现场办公,未提及弹性工作,生活化动机一般。

工作模式仅现场办公
办公地点市区核心地段
加班情况未提及(无法判断)

使命价值

70中等

机器人AI行业属于高速增长赛道,但实习岗位对社会的直接影响有限,意义感中等。

行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
Watch Jobs