工作职责:

-负责昆仑AI芯片接入PaddlePaddle、Pytorch、Tensorflow等深度学习框架和性能优化工作
-负责主流AI模型对昆仑AI芯片的适配和性能优化工作
-探索昆仑AI芯片分布式训练方案
-解决芯片互联、通讯库等关键问题

任职要求:

-计算机及计算机相关专业，硕士及以上学历
-熟悉C/C++，良好的编程习惯，熟悉计算机基础算法和数据结构，熟悉 Linux 开发环境
-有深度学习框架(Pytorch, Tensorflow, PaddlePaddle等)开发或使用经验优先
-精通MPI，对RDMA，GPU direct等技术有了解优先