職位描述
崗位職責(zé):1、參與國產(chǎn)芯片接入深度學(xué)習(xí)訓(xùn)練與推理框架的開發(fā)工作2、運(yùn)用編譯、融合算子等多種技術(shù)手段,加速國產(chǎn)芯片上的大模型推理3、其他相關(guān)前沿技術(shù)的調(diào)研與運(yùn)用崗位要求:1、具有良好的編程習(xí)慣,熟練使用Python/C++ 語言,有較強(qiáng)的debug能力;2、熟悉大模型推理算法,熟悉一種或者多種推理框架(vllm, lmdeploy, sglang, lightllm等)3、對(duì)深度學(xué)習(xí)框架有較深理解與實(shí)踐(包含不限于深度學(xué)習(xí)框架:Pytorch,TF,JAX)4、較強(qiáng)的自驅(qū)動(dòng)力,對(duì)先進(jìn)技術(shù)有濃厚興趣;5、有較強(qiáng)的團(tuán)隊(duì)精神和溝通交流能力;加分項(xiàng)1、熟悉英偉達(dá)或者國產(chǎn)硬件的編程模式2、對(duì)深度學(xué)習(xí)編譯器有較深理解以及擁有實(shí)際場(chǎng)景中的應(yīng)用經(jīng)驗(yàn)(torch.compile, triton,XLA,MLIR,TVM,Mojo等等)
企業(yè)介紹
上海人工智能實(shí)驗(yàn)室作為新型研發(fā)機(jī)構(gòu),于2020年7月在世界人工智能大會(huì)正式揭牌。實(shí)驗(yàn)室面向上??苿?chuàng)中心建設(shè)需求,開展戰(zhàn)略性、原創(chuàng)性、前瞻性的科學(xué)研究與技術(shù)攻關(guān),突破人工智能的重要基礎(chǔ)理論和關(guān)鍵核心技術(shù),支撐我國人工智能產(chǎn)業(yè)實(shí)現(xiàn)跨越式發(fā)展,目標(biāo)建成國際一流的人工智能實(shí)驗(yàn)室,成為享譽(yù)全球的人工智能原創(chuàng)理論和技術(shù)的策源地。牽頭科學(xué)家為湯曉鷗、姚期智、陳杰等人工智能領(lǐng)域世界知名學(xué)者。