1.工作职责
(1)AI芯片接入PaddlePaddle、Pytorch、Tensorflow等深度学习框架和性能优化工作;
(2)负责主流AI模型对昆仑AI芯片的适配和性能优化工作;
(3)探索昆仑AI芯片分布式训练方案,解决芯片互联、通讯库等关键问题。
2.职责要求
(1)计算机及计算机相关专业本科或本科以上学历;
(2)熟悉C/C++,良好的编程习惯,熟悉计算机基础算法和数据结构,熟悉Linux开发环境。
3.加分项
(1)有深度学习框架(Pytorch, Tensorflow, PaddlePaddle等)开发或使用经验者优先;
(2)熟悉MLPerf,有相关使用经验者优先;
(3)熟悉MPI、RDMA、GPU direct等技术者优先。
4.工作地点
北京市