岗位描述
岗位描述:
1、使用类C++语言,参与深度学习高性能算子库的开发(如激活、卷积、矩阵乘法、FlashAttention等)及其融合算子的实现。
2、负责算子的正确性与精度对齐,对算子性能进行极致调优,提升产品质量与泛化性,保障代码的可维护性与开发效率。
任职要求:
1、2027届硕士及以上学历,计算机、软件开发、系统优化等相关专业背景(有实习转正机会)。
2、具备良好的 C/C++ 编程能力和动手实践能力,算法基础扎实,对编译/运行时原理、计算机体系结构等有深入理解。
3、了解 GPU 等异构加速卡的体系架构及软件栈组成等基本概念。
4、了解代码优化的基本技术,比如向量化、双缓冲、指令流水、高速缓存等。
加分项:
1、接触过并行计算相关技术,例如 OpenMP、MPI、CUDA、pthread 或 athread 等。
2、在校期间有过深度学习或科学计算应用优化的项目经历。
3、对 Transformer 等知名模型结构有一定了解,并尝试过相关的优化工作。