崗位職責(zé)
1、使?類C++語(yǔ)言,開(kāi)發(fā)深度學(xué)習(xí)高性能算?庫(kù)的算?,?如激活算?、卷積算?、矩陣乘法、FlashAttention等,及其融合算子;
2、對(duì)算?的正確性和精度進(jìn)?對(duì)齊,對(duì)性能進(jìn)行極致調(diào)優(yōu),提升產(chǎn)品質(zhì)量,增強(qiáng)產(chǎn)品泛化性和可維護(hù)性,提高開(kāi)發(fā)效率。
任職資格
1、碩士及以上學(xué)歷,計(jì)算機(jī)、軟件開(kāi)發(fā)、系統(tǒng)優(yōu)化等計(jì)算機(jī)專業(yè)背景;
2、有良好的C/C++語(yǔ)?編程能力和動(dòng)手實(shí)踐能力,在算法基礎(chǔ),編譯/運(yùn)行時(shí)原理、體系結(jié)構(gòu)等基礎(chǔ)扎實(shí)牢靠;
3、了解GPU等異構(gòu)加速卡的體系架構(gòu),軟件棧組成等基本概念,有CUDA算子開(kāi)發(fā)經(jīng)驗(yàn)者優(yōu)先;
4、了解代碼優(yōu)化的基本技術(shù),?如向量化、雙緩沖、指令流?、?速緩存等;
5、(加分)有并?計(jì)算經(jīng)驗(yàn)優(yōu)先,?如OpenMP/MPI/CUDA/pthread/athread等;
6、(加分)有深度學(xué)習(xí)/科學(xué)計(jì)算應(yīng)用優(yōu)化經(jīng)驗(yàn)優(yōu)先,熟悉transformer等知名模型結(jié)構(gòu)者優(yōu)先。