【26届校招】算子库开发工程师-上海
上海
电子信息类·计算机类
使用简历深度优化功能，快速提升简历质量
职位介绍
招聘部门：研发
【岗位职责】
1、基于类 cuda 环境, 设计并实现高性能算子, 例如 gemm, Attention, FFN, CNN 等.
2、结合芯片微架构特性进行极致调优, 优化核心算子性能逼近理论设计上限.
3、开发性能评估工具, 并总结性能测试报告.
4、编写算子设计文档与交付文档，保障算子库的稳定性与可维护性。
【任职要求】
1、精通 C/C++, 熟悉 Linux 开发环境, 掌握 CMake, Git 等工程化工具, 必须具备良好的代码风格与调试能力.
2、深入理解 CPU/GPU 并行计算原理(如 SIMD/SIMT 架构), 有 CUDA/OpenCL/OpenMP 等编程经验者优先, 能独立完成简单算子的汇编实现者优先.
3、扎实的线性代数(矩阵运算, 特征分解), 数值分析(精度优化、数值稳定性)基础, 了解深度学习基本算子(卷积, 池化, 矩阵乘法等)的数学模型.
4、有独立进行过大模型算子优化相关工作成果者优先.
