職位描述

職位描述
基于NPU平臺,開發(fā)高性能計算庫,支撐CV,NLP,大模型等模型運行,包括:
1.完成高性能算子的需求分析與分解,滿足NPU的各種應(yīng)用場景;
2.與芯片/架構(gòu)團隊緊密配合,持續(xù)優(yōu)化算子性能,逐步完善軟硬件架構(gòu);
3.完成各類算子的軟件方案設(shè)計及開發(fā);
4.分析算子/網(wǎng)絡(luò)的性能瓶頸,提出性能優(yōu)化策略,達成性能目標;
職位要求
1.圖像處理、模式識別、機器學(xué)習(xí)等相關(guān)專業(yè)碩士及以上學(xué)歷;
2.精通C /Python軟件編程語言;
3.熟悉Onnx/TensorFLow/Pytorch一種主流神經(jīng)網(wǎng)絡(luò)框架,熟悉各類算子;
4.熟悉并行/異構(gòu)計算體系結(jié)構(gòu),在TPU/NPU/GPU一種平臺有2年以上的開發(fā)和優(yōu)化經(jīng)驗;
5.具備業(yè)界常用高性能庫(TensorRT/cuDNN/CUDA/OPENCV)開發(fā)和性能優(yōu)化經(jīng)驗者優(yōu)先;
6.熟悉指令流水,計算機體系結(jié)構(gòu);
7.良好的團隊溝通能力、技術(shù)攻關(guān)能力,責(zé)任心強,具備一定抗壓能力。
工作地點
地址:北京海淀區(qū)北京-海淀區(qū)海淀區(qū)


職位發(fā)布者
HR
小米科技有限責(zé)任公司


-
通信/電信/網(wǎng)絡(luò)設(shè)備/增值服務(wù)
-
1000人以上
-
私營·民營企業(yè)
-
北京市海淀區(qū)清河中街68號華潤五彩城寫字樓
相似職位
-
應(yīng)城長白班縫合包覆 面議應(yīng)城市 應(yīng)屆畢業(yè)生 不限武漢景行英才人力資源服務(wù)有限公司
-
行政接待崗 6000元以下應(yīng)城市 2年以上 大專三棵樹涂料股份有限公司
-
高級農(nóng)藝師 面議應(yīng)城市 3年以上 本科湖北富邦科技股份有限公司
-
NC業(yè)務(wù)專員(昆明) 5000-8000元應(yīng)城市 應(yīng)屆畢業(yè)生 不限紅葉
-
VIP客戶維護雙休8K不陌推 10000-15000元應(yīng)城市 應(yīng)屆畢業(yè)生 大專中國人壽保險股份有限公司濟南市分公司
-
沌口招聘專員五險一金月薪6-8K 面議應(yīng)城市 應(yīng)屆畢業(yè)生 不限湖北重點人力資源有限公司