返回岗位列表

深度学习框架开发工程师

查看原岗位

地点：上海薪资：薪资未公开日期：2026-01-29

岗位摘要

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

岗位职责

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

任职要求

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