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

薪资面议
硕士
上海市闵行区陈行公路2388号浦江科技广场16号楼M层
2024-11-05
 什么是官网闪投?
简历直投官网 无需重复填写简历 投后必反馈 进度实时更新 安全可靠官网可查 海量岗位5w+ 移动端投递方便

角色和期望:


1. 基于壁仞GPU生态,设计和实现深度学习训练/推理框架的核心模块,分析及优化框架的性能。


2. 探索和研究框架的前沿技术及发展方向,能够将新的技术应用到壁仞GPU的生态,促进壁仞GPU软件生态的成熟。


3. 解决日常测试中框架相关问题,并协同上下游团队分析定位壁仞软件栈的问题。



人才画像:


1. 计算机,自动化,模式识别、人工智能、电子信息等相关专业,5年以上深度学习领域工作经验。


2. 有良好的编程习惯与软件工程思想,有大型软件开发经验。熟练使用C/C++、Python等常见编程语言,熟悉常用的调试技巧与工具。


3. 掌握至少一种深度学习框架(Pytorch/Tensorflow/Mxnet/Oneflow/PaddlePaddle),至少在一种框架上有相关开发经验,对相关开源框架社区有贡献者优先。


4. 有 CUDA、OpenCL、OpenMP 等并行编程经验的优先。


5. 有深度学习算法研究与应用、深度学习模型的开发与适配相关经验者优先。


6. 熟悉AI编译原理,有XLA/Triton/TVM等编译优化经验者优先。