AI计算库开发工程师(J12749)

职位描述

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流程者优先。