高性能计算库研发工程师

岗位职责：
岗位描述：
1、使用类C++语言，参与深度学习高性能算子库的开发（如激活、卷积、矩阵乘法、FlashAttention等）及其融合算子的实现。
2、负责算子的正确性与精度对齐，对算子性能进行极致调优，提升产品质量与泛化性，保障代码的可维护性与开发效率。
任职资格：
任职要求：
1、2027届本科及以上学历，计算机、软件开发、系统优化等相关专业背景。
2、具备良好的 C/C++ 编程能力和动手实践能力，算法基础扎实，对编译/运行时原理、计算机体系结构等有深入理解。
3、了解 GPU 等异构加速卡的体系架构及软件栈组成等基本概念，有 CUDA 算子开发经验者优先。
4、了解代码优化的基本技术，比如向量化、双缓冲、指令流水、高速缓存等。
加分项：
1、有并行计算经验，比如熟悉 OpenMP/MPI/CUDA/pthread/athread 等。
2、有深度学习或科学计算应用优化经验优先，熟悉 Transformer 等知名模型结构者。
Base：无锡/北京/上海
具备行业竞争力的薪资待遇，六险一金