寒武纪|PyTorch框架研发实习生-27届

寒武纪 · 上海/北京 · 更新时间:2026-09-08

岗位职责

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

任职要求

任职条件: 1. 硕士及以上学历,计算机相关专业,2027届毕业生; 2. 熟练掌握C/C++/Python, 有扎实的编程基础, 良好的编程风格和工作经验; 3. 参与或负责过大型软件设计和性能调优或者开源社区经验;熟悉相关基础算法; 4. 良好的问题分析和解决能力 、沟通协作能力、领导能力和团队协作能力; 5. 熟悉框架内部机制(PyTorch或TensorFlow等), 具备丰富的框架开发和调试经验者优先; 6. 具备并行程序开发经验, 熟悉Pthread, CUDA, OpenCL等高性能编程者优先; 7.至少实习6个月,每周出勤4天及以上。

寒武纪招聘实习生:PyTorch框架研发实习生-27届岗位来自寒武纪招聘官网,具体内容以寒武纪招聘官网为准。