職位描述

職位描述
1. 為平頭哥ai芯片設(shè)計開發(fā)高性能互聯(lián)通訊庫。
2. 基于硬件特性進行極致性能優(yōu)化,并適配各種主流大模型和框架。
3. 深度分析各種實際應(yīng)用場景,發(fā)明和改進能適配硬件的拓撲結(jié)構(gòu)和加速算法。
4. 和其他團隊緊密合作,影響芯片架構(gòu)、編程模型和軟件平臺的設(shè)計和演進。
職位要求
1. 計算機/電子/通信/數(shù)學(xué)相關(guān)專業(yè)碩士及以上學(xué)歷,以及3年以上ai或半導(dǎo)體軟件開發(fā)相關(guān)從業(yè)經(jīng)驗。
2. 熟練掌握c/c++,具備優(yōu)秀的程序開發(fā)、設(shè)計和調(diào)試能力。
3. 熟練掌握各種基本算法,對操作系統(tǒng)、計算機體系結(jié)構(gòu)有深入的理解。
4. 滿足以下條件者從優(yōu):
- 良好的并行編程基礎(chǔ):有cuda/opencl/openmp/openmpi/openacc相關(guān)項目經(jīng)歷;
- 了解nccl、shmem或mpi等集合通信編程模型并參于過相關(guān)項目開發(fā);
- 熟悉allreduce、allgather及alltoall等典型集合運算操作的算法;
- 熟悉典型大模型訓(xùn)練框架像megatron、deepspeed或hovold等,并有過其中至少一個的使用及性能調(diào)優(yōu)經(jīng)驗;
- 深入?yún)⑴c過軟件項目的性能分析,并有一定的調(diào)優(yōu)經(jīng)驗;
- 有可編程芯片(如gpu/npu/tpu)架構(gòu)設(shè)計相關(guān)從業(yè)經(jīng)歷;
5. 良好的自我驅(qū)動能力及文檔梳理能力
工作地點
地址:杭州余杭區(qū)杭州文一西路969號


職位發(fā)布者
HR
阿里巴巴(中國)有限公司


-
IT服務(wù)·系統(tǒng)集成
-
1000人以上
-
中外合資(合資·合作)
-
杭州濱江區(qū)阿里巴巴濱江園區(qū)699
相似職位
-
電控工程師 11000-16000元應(yīng)屆畢業(yè)生 本科泰瑞機器股份有限公司
-
合成工藝研究員 10000-22000元拱墅區(qū) 應(yīng)屆畢業(yè)生 碩士華東醫(yī)藥股份有限公司
-
工程管理 面議西湖區(qū) 應(yīng)屆畢業(yè)生 不限浙江省武林建筑裝飾集團有限公司
-
副行長 20000-30000元余杭區(qū) 應(yīng)屆畢業(yè)生 本科寧波銀行股份有限公司杭州分行
-
方案設(shè)計技師 6000-12000元西湖區(qū) 應(yīng)屆畢業(yè)生 大專杭州美齊科技有限公司
-
品牌大使 3000-4000元拱墅區(qū) 應(yīng)屆畢業(yè)生 不限浙江金拱門食品有限公司