AI框架/图编译工程师（2027届）
岗位摘要
负责PyTorch框架在寒武纪芯片上的适配、优化与持续迭代；参与PyTorch社区开发，跟踪社区最新进展，完善和改进Cambricon Torch；优化PyTorch框架在寒武纪软件栈上的运行速度，提高模型训练和推理效率
岗位职责
- 负责PyTorch框架在寒武纪芯片上的适配、优化与持续迭代
- 参与PyTorch社区开发，跟踪社区最新进展，完善和改进Cambricon Torch
- 优化PyTorch框架在寒武纪软件栈上的运行速度，提高模型训练和推理效率
- 开发PyTorch框架相关配套工具，协助客户完成精度调试和性能调优
- 开发并优化高性能算子
任职要求
- 熟练掌握C/C++及Python，具备扎实的编程基础、良好的编程风格和工作习惯
- 有参与或主导大型软件设计、性能调优或开源社区贡献的经验
- 熟悉相关基础算法
- 具备良好的问题分析与解决能力、沟通协作能力及团队领导能力
- 熟悉PyTorch或TensorFlow等深度学习框架内部机制，具备丰富框架开发和调试经验者优先
- 具备并行程序开发经验，熟悉Pthread、CUDA、OpenCL等高性能编程技术者优先
- 学历要求：硕士及以上，在校/应届生
福利待遇
- 薪资范围：20,000–35,000元/月，共15薪
- 工作地点：北京市海淀区，北京航空航天大学学院路校区致真大厦D座
工作地址
北京海淀区北京航空航天大学学院路校区致真大厦D座