岗位描述
太初元碁是国家级专精特新“小巨人”与国家高新技术企业,核心团队成员始脱于国家超级计算无锡中心与清华大学,是国内AI算力底座的坚定探索者,坚持走异构众核的差异化技术路线,核心骨干多次荣获“戈登·贝尔奖”。首款自研AI加速卡T100已成功量产并规模商用,全面适配DeepSeek、Qwen(通义千问)、GLM(智谱)等多个主流AI大模型,用真实算力驱动AI未来。
岗位描述:
1、使用类C++语言,参与深度学习高性能算子库的开发(如激活、卷积、矩阵乘法、FlashAttention等)及其融合算子的实现。
2、负责算子的正确性与精度对齐,对算子性能进行极致调优,提升产品质量与泛化性,保障代码的可维护性与开发效率。
任职要求:
1、2027届硕士及以上学历,计算机、软件开发、系统优化等相关专业背景(有实习转正机会)。
2、具备良好的 C/C++ 编程能力和动手实践能力,算法基础扎实,对编译/运行时原理、计算机体系结构等有深入理解。
3、了解 GPU 等异构加速卡的体系架构及软件栈组成等基本概念。
4、了解代码优化的基本技术,比如向量化、双缓冲、指令流水、高速缓存等。
加分项:
1、接触过并行计算相关技术,例如 OpenMP、MPI、CUDA、pthread 或 athread 等。
2、在校期间有过深度学习或科学计算应用优化的项目经历。
3、对 Transformer 等知名模型结构有一定了解,并尝试过相关的优化工作。