返回岗位列表

PyTorch 框架研发实习生

查看原岗位

地点：上海薪资：280-350元/天日期：2026-03-03

岗位摘要

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

岗位职责

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

任职要求

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

工作地址

上海浦东新区展想广场祖冲之路2290号展想广场1805室