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

AI 摘要智能生成

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

职位描述

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

任职要求

• 熟练掌握C/C++/Python, 有扎实的编程基础, 良好的编程风格和工作经验；
• 参与或负责过大型软件设计和性能调优或者开源社区经验；
• 熟悉相关基础算法；
• 良好的问题分析和解决能力 、沟通协作能力、领导能力和团队协作能力；
• 熟悉框架内部机制(PyTorch或TensorFlow等), 具备丰富的框架开发和调试经验者优先；
• 具备并行程序开发经验, 熟悉Pthread, CUDA, OpenCL等高性能编程者优先。
• 注：本岗位设置笔试环节，请留意邮件或短信通知。