返回岗位列表

PyTorch框架研发实习生

查看原岗位

地点：北京薪资：280-350元/天日期：2026-03-03

岗位摘要

负责PyTorch框架在寒武纪芯片上的适配、优化与迭代；参与PyTorch社区开发，跟踪社区最新进展，完善和改进Cambricon Torch；优化PyTorch框架在寒武纪软件上的运行速度，提高模型训练和推理效率

岗位职责

负责PyTorch框架在寒武纪芯片上的适配、优化与迭代
参与PyTorch社区开发，跟踪社区最新进展，完善和改进Cambricon Torch
优化PyTorch框架在寒武纪软件上的运行速度，提高模型训练和推理效率
开发PyTorch框架相关配套工具，协助客户完成精度调试和性能调优
开发、优化高性能算子

任职要求

熟练掌握C/C++/Python，有扎实的编程基础，良好的编程风格和工作经验
参与或负责过大型软件设计和性能调优，或有开源社区经验
熟悉相关基础算法
良好的问题分析和解决能力、沟通协作能力、领导能力和团队协作能力
熟悉框架内部机制（PyTorch或TensorFlow等），具备丰富的框架开发和调试经验者优先
具备并行程序开发经验，熟悉Pthread、CUDA、OpenCL等高性能编程者优先
至少实习6个月，每周出勤4天及以上

工作地址

北京海淀区北京航空航天大学学院路校区致真大厦D座