AI互联通信软件专家70k以上 · 16薪
浙江硕士及以上5-10年通讯库,C++、计算机体系结构有基础,并行编程CUDA/OpenCL/OpenMP/OpenMPI/OpenACC相关项目经历,集合通信编程调优有经验,集合运算算法有相关经验
意外险#定期体检#五险一金#补充医疗保险#年终奖#绩效奖金#股票期权#员工旅游#带薪年假#住房补贴
职位描述
1. 为AI芯片设计开发高性能互联通讯库。
2. 基于硬件特性进行极致性能优化,并适配各种主流大模型和框架。
3. 深度分析各种实际应用场景,发明和改进能适配硬件的拓扑结构和加速算法。
4. 和其他团队紧密合作,影响芯片架构、编程模型和软件平台的设计和演进。
职位要求
1. 计算机/电子/通信/数学相关专业硕士及以上学历,以及3年以上AI或半导体软件开发相关从业经验。
2. 熟练掌握C/C++,具备优秀的程序开发、设计和调试能力。
3. 熟练掌握各种基本算法,对操作系统、计算机体系结构有深入的理解。
4. 满足以下条件者从优:
- 良好的并行编程基础:有CUDA/OpenCL/OpenMP/OpenMPI/OpenACC相关项目经历;
- 了解NCCL、SHMEM或MPI等集合通信编程模型并参于过相关项目开发;
- 熟悉AllReduce、AllGather及AlltoAll等典型集合运算操作的算法;
- 熟悉典型大模型训练框架像Megatron、DeepSpeed或Hovold等,并有过其中至少一个的使用及性能调优经验;
- 深入参与过软件项目的性能分析,并有一定的调优经验;
- 有可编程芯片(如GPU/NPU/TPU)架构设计相关从业经历;
5. 良好的自我驱动能力及文档梳理能力
base:杭州 杭州市余杭区向往街1122号欧美金融城(EFC)英国中心西楼T6