锦鲤求职

寒武纪

AI框架/图编译工程师-27届

锦鲤会替你打开官网网申、按简历自动填表提交,你只需要在关键步骤确认。

岗位描述

岗位职责:

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

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

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

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

开发、优化高性能算子。

任职要求:

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

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

熟悉相关基础算法;

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

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

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

注:本岗位设置笔试环节,请留意邮件或短信通知。

人工智能方向的申请准备

先核对岗位要求与自己的经历,再准备档案、投递和面试。

阅读求职步骤与示例 →