Deep Learning Solution Architect--HPC 35-65K·13薪
  • 上海
  • 經(jīng)驗不限
  • 本科

職位描述:

  • 機器學(xué)習
  • 模型加速/性能優(yōu)化
  • C/C++
  • HPC
  • CUDA

工作方向為加速計算,部門(mén)主要工作內容為GNN加速、推理加速、性能仿真/預測等。歡迎對高性能計算、系統設計、加速訓練等方向感興趣的同學(xué)投遞簡(jiǎn)歷。日常工作內容包括但是不限于 1. 針對架構和算法特征開(kāi)發(fā)和優(yōu)化CUDA核函數代碼 2. 對深度學(xué)習框架進(jìn)行優(yōu)化,包括系統設計,并行算法,網(wǎng)絡(luò )通信等方面 3. 對常見(jiàn)的深度學(xué)習workload建模,評估或預測在各種環(huán)境下的運行時(shí)間 4. GNN相關(guān)的優(yōu)化,例如從圖劃分、采樣、以及embedding存儲相關(guān)角度進(jìn)行的優(yōu)化 【技能需求】 職位需求是對AI/DL技術(shù)有一定了解,包括但不限于算法優(yōu)化、推理加速、并行計算、性能仿真、以及其他HPC方向,最好比較熟悉C++,接觸過(guò)CUDA,或有一些系統優(yōu)化方面的經(jīng)驗。

張先生

張先生 3日內活躍

NVidia · HR
工作地址:

上海浦東新區矽岸國際

查看更多信息
更新時(shí)間:2024-07-21