新業(yè)務(wù)部-AI高性能算子開(kāi)發(fā)工程師/專家
28000-55000元
應(yīng)屆畢業(yè)生
本科



- 全勤獎(jiǎng)
- 節(jié)日福利
- 不加班
- 周末雙休
職位描述

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


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


-
通信/電信/網(wǎng)絡(luò)設(shè)備/增值服務(wù)
-
1000人以上
-
私營(yíng)·民營(yíng)企業(yè)
-
北京市海淀區(qū)清河中街68號(hào)華潤(rùn)五彩城寫(xiě)字樓
相似職位
-
WEB前端中級(jí)工程師 10000-16000元浦口區(qū) 應(yīng)屆畢業(yè)生 本科鈦能科技股份有限公司
-
嵌入式軟件助理工程師 8000-13000元浦口區(qū) 應(yīng)屆畢業(yè)生 本科鈦能科技股份有限公司
-
采購(gòu)內(nèi)勤文員 面議浦口區(qū) 應(yīng)屆畢業(yè)生 不限南京交韻汽車科技有限公司
-
市場(chǎng)廣培生 10000-15000元浦口區(qū) 應(yīng)屆畢業(yè)生 本科中國(guó)平安人壽保險(xiǎn)股份有限公司江蘇分公司
-
施工員 5000-8000元浦口區(qū) 應(yīng)屆畢業(yè)生 大專江蘇東道交通科技集團(tuán)有限公司
-
客戶開(kāi)發(fā)專員 面議浦口區(qū) 應(yīng)屆畢業(yè)生 不限焦點(diǎn)教育科技有限公司