返回岗位列表

寒武纪

软件开发工程师(C++)

地点:上海 薪资:25-50K 日期:2026-03-03

岗位摘要

优化Torch Compile模式下各类神经网络的性能,包括网络性能瓶颈分析、编译耗时优化、codegen性能优化等;紧跟PyTorch社区最新技术进展,及时将前沿性能优化技术迁移到MLU平台,保持技术竞争力和创新性

岗位职责

  • 优化Torch Compile模式下各类神经网络的性能,包括网络性能瓶颈分析、编译耗时优化、codegen性能优化等
  • 紧跟PyTorch社区最新技术进展,及时将前沿性能优化技术迁移到MLU平台,保持技术竞争力和创新性

任职要求

  • 有较强的C++和Python功底,有扎实的数据结构和算法基础
  • 学习能力强,有强烈的求知欲和自我驱动能力,良好的沟通能力和团队协作能力
  • 加分项:熟悉PyTorch/TensorFlow等主流深度学习框架,熟悉框架内部的模块、机制
  • 加分项:熟悉torch inductor、Triton、XLA、MLIR、TVM等深度学习编译技术

加分项

  • 1:熟悉PyTorch/TensorFlow等主流深度学习框架,熟悉框架内部的模块、机制
  • 2:熟悉torch inductor、Triton、XLA、MLIR、TVM等深度学习编译技术
  • 陶先生 刚刚活跃

工作地址

上海浦东新区展想广场1号楼1805