返回岗位列表

寒武纪

ai框架研发工程师(流量/平方英寸)

地点:北京 薪资:30-60K 日期:2026-03-03

岗位摘要

优化TensorFlow下各类神经网络的性能,包括网络性能瓶颈分析;进行框架内核开发与性能优化工作;紧跟社区最新技术进展,及时将前沿性能优化技术迁移到MLU平台;保持技术竞争力和创新性;有很强的C++和Python功底,有扎实的数据结构和算法基础

岗位职责

  • 优化TensorFlow下各类神经网络的性能,包括网络性能瓶颈分析
  • 进行框架内核开发与性能优化工作
  • 紧跟社区最新技术进展,及时将前沿性能优化技术迁移到MLU平台
  • 保持技术竞争力和创新性

任职要求

  • 有很强的C++和Python功底,有扎实的数据结构和算法基础
  • 学习能力强,有强烈的求知欲和自我驱动能力
  • 加分项:熟悉TensorFlow等主流深度学习框架,熟悉框架内部的模块、机制
  • 加分项:对TensorFlow异步执行器(executor)、内存分配器(如BFC allocator)、设备运行时接口(如CUDA runtime)、图优化器(Grappler)有深度了解和实际开发经验
  • 加分项:了解PyTorch、Paddle等其他深度学习框架
  • 加分项:熟悉XLA、MLIR深度学习编译技术
  • 加分项:对搜索广告推荐领域相关模型优化、技术发展有一定的实际经验

加分项

  • 1:熟悉TensorFlow等主流深度学习框架,熟悉框架内部的模块、机制
  • 对tensorflow异步执行器(executor)、内存分配器(如bfc allocator)、设备运⾏时接口(如cuda runtime)、图优化器(grappler)有深度的了解和实际的开发经验
  • 如果了解其他框架,如:pytorch、paddle,亦可
  • 2:熟悉XLA、MLIR深度学习编译技术
  • 3:对搜索广告推荐领域相关模型优化、技术发展有一定的实际经验
  • 芦先生 刚刚活跃

工作地址

北京海淀区北航致真大厦D座