岗位描述
岗位职责:
1、使⽤类C++语言,开发深度学习高性能算⼦库的算⼦,⽐如激活算⼦、卷积算⼦、矩阵乘法、FlashAttention等,及其融合算子;
2、对算⼦的正确性和精度进⾏对齐,对性能进行极致调优,提升产品质量,增强产品泛化性和可维护性,提高开发效率。
任职资格:
1、硕士及以上学历,计算机、软件开发、系统优化等计算机专业背景;
2、有良好的C/C++语⾔编程能力和动手实践能力,在算法基础,编译/运行时原理、体系结构等基础扎实牢靠;
3、了解GPU等异构加速卡的体系架构,软件栈组成等基本概念,有CUDA算子开发经验者优先;
4、了解代码优化的基本技术,⽐如向量化、双缓冲、指令流⽔、⾼速缓存等;
5、(加分)有并⾏计算经验优先,⽐如OpenMP/MPI/CUDA/pthread/athread等;
6、(加分)有深度学习/科学计算应用优化经验优先,熟悉transformer等知名模型结构者优先。