職位描述

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


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


-
IT服務(wù)·系統(tǒng)集成
-
1000人以上
-
中外合資(合資·合作)
-
杭州濱江區(qū)阿里巴巴濱江園區(qū)699
相似職位
-
涂裝工程師 9000元以上臨平區(qū) 應(yīng)屆畢業(yè)生 不限上海佩琪信息技術(shù)有限公司
-
直播場控 7000-10000元蕭山區(qū) 應(yīng)屆畢業(yè)生 大專杭州凌巨網(wǎng)絡(luò)科技有限公司
-
項(xiàng)目管理工程師 面議臨平區(qū) 應(yīng)屆畢業(yè)生 不限杭州福斯達(dá)深冷裝備股份有限公司
-
招聘專家 40000-60000元西湖區(qū) 應(yīng)屆畢業(yè)生 本科中徽機(jī)電科技股份有限公司
-
保險(xiǎn)經(jīng)理人 面議應(yīng)屆畢業(yè)生 不限中國人壽保險(xiǎn)股份有限公司
-
質(zhì)量文員 5000-7000元濱江區(qū) 應(yīng)屆畢業(yè)生 不限杭州銳健馬斯汀醫(yī)療器材有限公司