C++開發(fā)實(shí)習(xí)生(機(jī)器學(xué)習(xí)框架方向)
適用專業(yè)
計(jì)算機(jī)
適用年級(jí)
高年級(jí)
截止時(shí)間
2023-06-21
工作地區(qū)
北京
公司介紹
崗位職責(zé)
1、參與公司機(jī)器學(xué)習(xí)平臺(tái)相關(guān)組件的開發(fā)和維護(hù),包括GPU、CPU分布式訓(xùn)練、GPU推理優(yōu)化等;
2、支持業(yè)務(wù)算法團(tuán)隊(duì)進(jìn)行模型效果和訓(xùn)練性能優(yōu)化;
崗位要求
1、本科及以上學(xué)歷優(yōu)先,具備扎實(shí)的計(jì)算機(jī)基礎(chǔ);
2、熟悉C、C++、Python編程語(yǔ)言,有CUDA相關(guān)編程經(jīng)驗(yàn)優(yōu)先;
3、了解TensorFlow、Pytorch、DeepSpeed、Megatron等開源框架,有源碼或相關(guān)實(shí)習(xí)經(jīng)驗(yàn)優(yōu)先;
4、了解推理優(yōu)化相關(guān)技術(shù),有編譯優(yōu)化相關(guān)實(shí)踐(TensorRT、XLA、TVM等)優(yōu)先;
5、具備良好的團(tuán)隊(duì)溝通協(xié)作能力和學(xué)習(xí)能力,有責(zé)任心和執(zhí)行力。
投遞方式