返回岗位列表

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

查看原岗位

地点：上海薪资：40-60K 日期：2026-05-08

岗位摘要

基于壁仞GPU生态，设计和实现深度学习训练/推理框架的核心模块，分析及优化框架性能；探索和研究框架的前沿技术及发展方向，将新技术应用到壁仞GPU生态中；推动壁仞GPU软件生态的成熟与完善

岗位职责

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

任职要求

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

工作地址

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