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