太初（杭州）集成电路有限公司

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