职位描述
1、负责开发针对GPU平台的AI算子库,并构建完整的测试集进行正确性和性能验证;
2、负责跟进AI社区流行的最新算法,将其在自研GPU平台上进行高性能实现与优化;
3、负责与硬件架构、编译器团队合作,对AI与通用计算的算法进行极致的性能优化;
4、负责定位软硬件的性能瓶颈,为软件栈的开发和新一代硬件的设计提供改进建议。任职要求
1、计算机相关专业,本科及以上学历, 3-5年相关工作经验;
2、熟悉C/C++编程,有良好的编程习惯和较强的问题解决能力;
3、熟悉至少一种GPU硬件体系架构及其编程模型;
4、熟悉至少一种异构计算编程体系,如:CUDA、HIP、SYCL、OpenCL等;
5、熟悉至少一种计算库的整体框架及开发流程,如DNN、BLAS、FFT、CUB等;
6、熟悉常用的并行计算算法,如GEMM,有对其进行性能分析和深度优化的经验;
7、熟悉常见的性能分析工具,如Nsight Compute、perf等;
8、有较强的团队协作能力与沟通能力,有团队和项目管理经验者优先;
9、对技术和代码品质有追求,有过敏捷开发经验,熟悉Scrum流程者优先。