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