【職位描述】
1. Cuda C Kernel算子編寫(xiě)優(yōu)化
2. 測(cè)試,驗(yàn)證國(guó)內(nèi)外的CuTile, TileLang , Triton等新語(yǔ)言
3. 部署算子到Jetson Orin等開(kāi)發(fā)平臺(tái),確認(rèn)可用性,以及反饋問(wèn)題
4. 同外部伙伴一起解決編寫(xiě)部署過(guò)程中遇到的問(wèn)題
5. 總結(jié)匯報(bào)自動(dòng)駕駛領(lǐng)域最新的算法進(jìn)展,成果。
6. 將SOTA的算法成果應(yīng)用到當(dāng)前算法和模型的改進(jìn)及優(yōu)化過(guò)程當(dāng)中。
【崗位要求】
1. 掌握python,熟悉pytorch框架,具備扎實(shí)的深度學(xué)習(xí)和大語(yǔ)言模型算法基礎(chǔ)。
2. 熟悉Transformer架構(gòu),模型量化方法,主動(dòng)學(xué)習(xí)mamba等新架構(gòu)。
3. 掌握kernel編寫(xiě)方法,會(huì)積極使用各種主流編譯庫(kù)CuTile,Triton等進(jìn)行算子編寫(xiě)。
4. 人工智能、計(jì)算機(jī)、信息、數(shù)學(xué)、電子、自動(dòng)化、機(jī)械、測(cè)控、汽車(chē)等相關(guān)專(zhuān)業(yè)碩士及以上學(xué)歷,具有扎實(shí)的數(shù)學(xué)功底和算法基礎(chǔ);計(jì)算機(jī)基礎(chǔ)扎實(shí),掌握C++、TensoRT、Linux(Ubuntu)等人工智能基礎(chǔ)軟件。
5. 英語(yǔ)6級(jí)水平,業(yè)務(wù)交流無(wú)障礙,同時(shí)日語(yǔ)能夠進(jìn)行日常會(huì)話(huà)。
6. 熟悉VAD,UniAD等開(kāi)源自動(dòng)駕駛算法優(yōu)先。具有1-2年相關(guān)工作經(jīng)驗(yàn)者優(yōu)先,新畢業(yè)生也可。