崗位職責:
1、 負責AI訓練芯片架構的探索和設計;
2、 負責AI訓練系統的軟硬件協同設計;
3、 負責不同業(yè)務場景下AI模型訓練流程中的關鍵任務和典型算子分析;
1、 熟悉計算機體系結構,對典型AI硬件加速器架構或者GPU架構有深入的了解;
2、 有AI模型優(yōu)化相關的經驗,對典型AI模型計算訪存行為有較好的理解;
3、 熟悉分布式訓練流程,熟悉方向傳播和參數更新過程;
4、 熟練掌握C/C++,python等相關編程語言;
加分項:
1、 計算機網絡的TCP/IP協議棧或者RDMA協議有了解;
2、 了解TensorFlow、Pytorch等深度學習框架;