深度学习框架工程师/架构师

查看原岗位

地点：上海薪资：30-60K 日期：2026-06-26

岗位摘要

基于壁仞GPU生态，设计和实现深度学习训练/推理框架的核心模块，分析及优化框架性能；探索和研究框架的前沿技术及发展方向，将新技术应用于壁仞GPU生态，促进软件生态成熟；解决日常测试中框架相关问题，协同上下游团队分析定位壁仞软件栈的问题

计算机、自动化、模式识别、人工智能、电子信息等相关专业，5年以上深度学习领域工作经验
有良好的编程习惯与软件工程思想，有大型软件开发经验，熟练使用C/C++、Python等常见编程语言，熟悉常用调试技巧与工具
掌握至少一种深度学习框架（Pytorch/Tensorflow/Mxnet/Oneflow/PaddlePaddle），至少在一种框架上有相关开发经验，对相关开源框架社区有贡献者优先
有CUDA、OpenCL、OpenMP等并行编程经验者优先
有深度学习算法研究与应用、深度学习模型的开发与适配相关经验者优先
熟悉AI编译原理，有XLA/Triton/TVM等编译优化经验者优先

上海闵行区临港浦江科技广场16号楼