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