寒武纪
PyTorch框架研发实习生-27届
PyTorch框架研发实习生-27届
发布于 大约 2 小时前实习/见习
上海市
无经验要求
实习生
仅现场办公
硕士
机器学习工程
Ai芯片
Cuda
Opencl
Pytorch
并行编程
深度学习框架
算法
高性能计算
AI 估算 · 6k–10k
上海AI芯片公司实习，技术含量高，硕士实习生月薪约6000-10000元，市场中等水平。
职位详情
关于这个职位
该实习岗位专注于PyTorch深度学习框架在寒武纪AI芯片上的适配、优化与性能调优
你将参与开源社区开发，跟踪最新进展，并开发高性能算子及配套工具
适合对AI基础设施底层技术有强烈兴趣的计算机专业研究生
最低要求
硕士及以上学历，计算机相关专业，2027届毕业生
熟练掌握C/C++/Python，有扎实的编程基础，良好的编程风格和工作经验
参与或负责过大型软件设计和性能调优或者开源社区经验
熟悉相关基础算法
良好的问题分析和解决能力、沟通协作能力、领导能力和团队协作能力
至少实习6个月，每周出勤4天及以上
工作职责
负责PyTorch框架在寒武纪芯片上适配、优化与迭代
参与PyTorch社区开发，跟踪社区最新进展，完善和改进Cambricon Torch
优化PyTorch框架在寒武纪软件上运行速度，提高模型训练和推理效率
开发PyTorch框架相关配套工具，协助客户完成精度调试和性能调优
开发、优化高性能算子
优先资格
熟悉框架内部机制(PyTorch或TensorFlow等)，具备丰富的框架开发和调试经验者优先
具备并行程序开发经验，熟悉Pthread, CUDA, OpenCL等高性能编程者优先
AI 洞察
优缺点分析
优点
- 寒武纪是国内AI芯片龙头企业，实习经历含金量高，为简历增色
- 接触PyTorch框架底层开发，技术深度大，能积累宝贵的底层优化经验
- 参与开源社区，有机会贡献代码，提升个人在行业内的知名度
- 需要同时掌握深度学习框架和芯片架构知识，学习曲线较陡峭
- 实习期至少6个月，每周出勤4天，时间投入较大，需平衡学业
- 工作中可能涉及大量性能调优和问题排查，需要较强自驱力和耐心
- 适合对AI基础设施、深度学习框架底层有浓厚兴趣，技术基础扎实，能够长期稳定实习的计算机专业研究生
缺点 / 挑战
暂无明显挑战项
角色解读
- 深入理解AI芯片与框架协同，成为AI基础设施领域的专家，可转向编译器或芯片软件栈研发
- 实习期表现优秀可转为正式员工，参与更核心的研发工作，如新一代框架适配
- 参与开源社区，积累影响力，未来可从事深度学习框架或高性能计算相关研究岗位
- 负责PyTorch框架在寒武纪芯片上的适配、优化与迭代，确保深度学习模型高效运行
- 参与PyTorch社区开发，跟踪最新进展，完善公司内部工具Cambricon Torch
- 优化训练和推理效率，开发高性能算子，并开发配套工具协助客户调试与性能调优
- 熟练掌握C/C++/Python，具备扎实的编程基础和良好的编程风格
- 熟悉PyTorch或TensorFlow等深度学习框架的内部机制，有丰富的框架开发经验者优先
- 具备并行程序开发经验，熟悉Pthread、CUDA、OpenCL等高性能编程技术
- 良好的问题分析与解决能力、沟通协作能力以及团队合作精神
申请策略
- 在求职信中表达对AI芯片+框架结合的理解，并强调能稳定实习6个月以上
- 展示技术热情和快速学习能力，可举例说明如何解决复杂技术问题
- 突出C/C++/Python编程项目经历，尤其是大型软件设计或性能调优相关
- 展示PyTorch或TensorFlow框架的使用、二次开发或源码阅读经历
- 如有并行编程（CUDA/OpenCL）或高性能计算项目经验，务必重点描述
- 强调开源贡献经历，如参与过开源项目或提交过PR
- 提前学习PyTorch源码，了解框架架构，特别是与设备交互的部分
- 复习CUDA编程和并行计算概念，熟悉常用优化技巧
面试指南
- 从基本原理出发，结合具体项目经验，用STAR法则描述问题、动作和结果
- 展示对性能指标的关注（如延迟、吞吐量、内存占用），并量化优化效果
- 体现系统性思维，说明优化的权衡（如精度与速度、开发成本与收益）
- 解释PyTorch自动求导机制和动态图的实现原理
- 如何优化一个PyTorch算子的性能？请举例说明
- 描述你参与过的C++项目中的性能调优经验，使用了哪些工具和方法？
- CUDA编程中如何优化内存访问以提高带宽利用率？
- 对寒武纪芯片架构有什么了解？你认为在AI芯片上优化框架的关键点是什么？
职位点评
63
综合评分
前沿AI框架研发实习，技术驱动，成长性极高，但需要现场办公且时间投入大。
从学习成长、工作节奏、岗位方向和实习待遇综合评估，方便比较实习机会。
更适合这类人
适合追求技术深度和快速成长，对短期薪资和WLB要求不高的学生。
表现最好
成长发展
相对薄弱
工作生活
薪资福利60
成长发展90
工作生活40
使命价值60
薪资福利
60中等
实习薪资处于市场中上水平，公司未明确福利，但作为上市公司基本保障应有。
薪资信号未披露（AI估算：6K-10K/月）
成长发展
90较高
技术前沿（PyTorch底层+AI芯片），参与开源社区，成长空间巨大，非常符合发展性动机。
技术前沿前沿/新兴技术
技术栈PyTorch、C++、CUDA、OpenCL、高性能计算
成长机会参与PyTorch社区开发、跟踪社区最新进展
业务类型profit_center
工作生活
40较低
需现场办公，每周出勤4天，但未提及加班情况，灵活性较低。
工作模式仅现场办公
办公地点未明确
加班情况未提及（无法判断）
使命价值
60中等
AI芯片行业高速增长，但实习岗位社会影响力一般，创新性较强。
行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
寒武纪 的其他在招职位
相似职位推荐
Watch Jobs