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