深度学习框架开发Leader65k-95k

上海经验10年以上本科及以上技术经理|主管
岗位所属职位类型
全职

  • 人工智能服务
  • 智能硬件
  • 云计算|云存储
燧原科技
分享到微信
微信扫一扫,用小程序打开分享
职位诱惑:

团队实力强

职位描述:

部门介绍:

燧原科技大编译(CLT)软件部门负责打造GCU AI芯片的软件栈核心组件。为使能GCU芯片,挖掘硬件潜能,优化模型性能提供可靠,可扩展,可移植的先进的产品解决方案。同时为用户提供全栈的性能分析,调优及功能调试的完整的工具链。下属框架(Framework)团队负责自研芯片和主流AI开源框架集成,模型图的转换及优化,同时负责与JIT和AOT算子对接以及GCU runtime API的调度及集成工作。

工作职责:该职位负责管理燧原框架组的日常工作和人事。和团队架构师制定团队的技术产品路线;和PL,PM一起合作交付项目目标。具体包括以下几项:

带领团队负责第三方框架:Tensorflow/PyTorch/ONNX/MindSpore/Paddle 等的对接方案和技术支持。扩展燧原芯片的框架支持生态。
带领团队分析和优化AI框架的性能,添加,维护框架层的优化pass(es)
带领团队负责AI框架AOT/JIT 编译功能支持,动态图和静态图的支持
带领团队针对神经网络计算设备,设计和调优AI框架图级别调度和融合等算法优化;
跟踪/分析/评估各主流开源深度学习框架
参与自研AI芯片系统软件栈的整体设计和实现,性能调优
和上下左右团队协同合作,拉通对齐需求和风险
职位要求:

• 计算机/电子信息类专业本科,硕士及以上学历

• 理解深度学习理论,熟悉开源AI框架的源码和技术特点:Tensorflow/PyTorch/ONNX/MindSpore/Paddle etc.

• 熟悉MLIR/LLVM, TVM 等AI编译栈技术。

• 在知名公司负责过框架组的技术工作和产品。

• 有扎实的C/C++/Python编程能力

• 对计算机体系结构有深刻理解

• 有较强的快速学习能力,良好的团队合作能力和沟通能力

• 流利的英语阅读、写作能力

• (加分项)有编译器(LLVM/GCC),MLIR框架经验

• (加分项)有CUDA, TensorRT, OpenCL或其他并行计算/HPC开发经验

• (加分项)有XLA,TVM,ONNX, PlaidML, Halide, IREE, Tiramisu and Glow计算图优化经验

• (加分项)有分布式计算框架的研发、运维、性能评测调优等经验

• (加分项)有在基于深度学习的目标识别、目标检测、机器视觉、NLP领域有过工程经验或发表过学术论文

附加信息:

  • 是否为管理岗:

工作地址

上海 - 浦东新区- 张江镇盛夏路61弄张润大厦查看地图

职位发布者:

拉勾安全提示
· 求职中如遇招聘方扣押证件、要求提供担保或收取财物、强迫入股或集资、收取不正当利益或其他违法情形,请立即举报
· 如遇岗位要求海外工作,请提高警惕,谨防诈骗
面试评价
【查看更多评价】
该职位尚未收到面试评价
上海燧原科技有限公司

燧原科技

相似职位