深度学习框架工程师/架构师
岗位摘要
基于壁仞GPU生态，设计和实现深度学习训练/推理框架的核心模块，分析及优化框架性能；探索和研究框架的前沿技术及发展方向，将新技术应用于壁仞GPU生态，促进软件生态成熟；解决日常测试中框架相关问题，协同上下游团队分析定位壁仞软件栈的问题
岗位职责
- 基于壁仞GPU生态，设计和实现深度学习训练/推理框架的核心模块，分析及优化框架性能
- 探索和研究框架的前沿技术及发展方向，将新技术应用于壁仞GPU生态，促进软件生态成熟
- 解决日常测试中框架相关问题，协同上下游团队分析定位壁仞软件栈的问题
任职要求
- 计算机、自动化、模式识别、人工智能、电子信息等相关专业，5年以上深度学习领域工作经验
- 有良好的编程习惯与软件工程思想，有大型软件开发经验，熟练使用C/C++、Python等常见编程语言，熟悉常用调试技巧与工具
- 掌握至少一种深度学习框架（Pytorch/Tensorflow/Mxnet/Oneflow/PaddlePaddle），至少在一种框架上有相关开发经验，对相关开源框架社区有贡献者优先
- 有CUDA、OpenCL、OpenMP等并行编程经验者优先
- 有深度学习算法研究与应用、深度学习模型的开发与适配相关经验者优先
- 熟悉AI编译原理，有XLA/Triton/TVM等编译优化经验者优先
福利待遇
- 薪资范围：30-60K·15薪
- 工作地点：上海闵行区临港浦江科技广场16号楼
工作地址
上海闵行区临港浦江科技广场16号楼