岗位描述
工作职责
1. 负责华为昇腾平台上基础AI算子的开发、适配和测试。
2. 根据现有模板或开发文档完成算子代码编写和功能验证。
3. 负责算子的编译、部署、单元测试和问题排查。
4. 协助完成模型中不支持算子的定位和基础适配工作。
5. 配合完成算子精度测试、性能测试和测试报告整理。
6. 维护算子开发文档、测试用例和相关代码。
7. 配合模型部署及性能优化人员解决算子相关问题。
工作要求
1. 计算机、软件工程、电子信息、自动化等相关专业,本科及以上学历。
2. 有扎实的C++编程能力。
3. 熟悉模板编程。
4. 了解Linux开发环境和常用命令。
5. 了解数据结构、内存管理、函数调用等基础知识。
6. 了解深度学习基本概念,知道卷积、激活函数、矩阵运算等常见操作。
7. 具备基本的代码阅读、问题分析和学习能力。
8. 能够按照开发文档和现有示例完成基础算子开发。
加分项
1. 接触过华为昇腾CANN、Ascend C、TBE或MindSpore。
2. 熟悉Tensorflow/Caffe/Pytorch/MXNet/PaddlePaddle等开源深度学习框架。
3. 有CUDA、GPU编程、并行计算或算子开发经验。
4. 熟悉CMake、Python等开发工具。
5. 开发过ReLU、Softmax、MatMul、Reduce等基础算子。
6. 了解算子精度对比、性能分析和Profiling工具。
专业要求
计算机专业