崗位職責(zé)
1. 負(fù)責(zé)智能超聲診療模型在多種不同硬件平臺(tái)的極致性能優(yōu)化,結(jié)合編譯優(yōu)化、并行計(jì)算優(yōu)化、圖融合、高效 CUDA 算子開(kāi)發(fā)實(shí)現(xiàn)行業(yè)領(lǐng)先的超聲設(shè)備端推理性能。
2. 針對(duì)特定NPU硬件計(jì)算平臺(tái),基于對(duì)硬件體系結(jié)構(gòu)的深入理解,實(shí)現(xiàn)對(duì)硬件的高效利用。
3. 針對(duì)Pytorch/CUDA相關(guān)GPU計(jì)算任務(wù),進(jìn)行算子和系統(tǒng)優(yōu)化,提升訓(xùn)練/推理效率。
崗位要求
1. 工作3年以上,計(jì)算機(jī)/數(shù)學(xué)/物理/電子工程/自動(dòng)控制專業(yè)碩士及以上學(xué)歷。
2. 優(yōu)秀的代碼能力,熟練掌握C/C++或Python,有 CUDA 開(kāi)發(fā)經(jīng)驗(yàn),熟悉TVM/TensorRT/Triton/Cutlass等推理框架優(yōu)先。
3. 熟悉計(jì)算機(jī)體系結(jié)構(gòu),對(duì)GPU、NPU等有比較深刻的理解。
4. 熟悉CNN/Transformers/DETR等主流模型原理。5. 良好的團(tuán)隊(duì)協(xié)作能力;創(chuàng)新性強(qiáng),有良好的動(dòng)手實(shí)現(xiàn)能力,對(duì)技術(shù)有熱情。