岗位描述
【岗位介绍】
该岗位负责基于自研芯片的高性能AI算子的设计和实现。算子是构成深度学习网络的基本单位,算子的性能直接影响到深度学习网络的整体性能。该岗位的工程师通过研究并行计算算法,结合自研芯片,实现高性能算子,并接入到AI主流框架中。具体包括:
-设计和实现AI数学库,如BLAS, MIOpen, DNN等
-分析Benchmark性能,提出性能瓶颈,并改进算子实现或提出方案
【基本要求】
1.计算机或相关专业硕士及以上学历,对GPU工作原理有一定理解;
2.熟练掌握C/C++和Python编程,有扎实的编程基础、良好的编程风格和工作习惯;
3.了解主流的CNN/RNN模型;
4.了解高性能算子自动生成方法;
【加分项目】
1.具有CUDA/Opencl/cuDNN开发经验优先;
2.熟悉常用AI库,如BLAS,FFT,MPI,DNN,MiOpen等优先
【学历要求】
硕士及以上