PyTorch框架研发实习生

薪资面议
祖冲之路2290号展想广场1号楼1805
2025-02-28
 什么是官网闪投?
简历直投官网 无需重复填写简历 投后必反馈 进度实时更新 安全可靠官网可查 海量岗位5w+ 移动端投递方便


岗位职责

1. 负责PyTorch框架在寒武纪芯片上适配、优化与迭代;

2. 参与PyTorch社区开发, 跟踪社区最新进展, 完善和改进Cambricon Torch;

3. 优化PyTorch框架在寒武纪软件上运行速度, 提高模型训练和推理效率;

4. 开发PyTorch框架相关配套工具, 协助客户完成精度调试和性能调优;

5. 开发、优化高性能算子。



任职资格
条件:

1. 熟练掌握C/C++/Python, 有扎实的编程基础, 良好的编程风格和工作经验;

2. 参与或负责过大型软件设计和性能调优或者开源社区经验;

3. 熟悉相关基础算法;

4. 良好的问题分析和解决能力 、沟通协作能力、领导能力和团队协作能力;

5. 熟悉框架内部机制(PyTorch或TensorFlow等), 具备丰富的框架开发和调试经验者优先;

6. 具备并行程序开发经验, 熟悉Pthread, CUDA, OpenCL等高性能编程者优先。

寒武纪
人工智能
上市
北京,上海,深圳,合肥,西安,南京
查看其他 11 个职位