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