Ivyea Jobs 8332 roles · 120 companies · 刚刚
芯动科技 · AI 芯片

AI计算库开发工程师

武汉 社招 · 全职 GPU岗位 工程开发

职位描述

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流程者优先。
芯动科技去官方页面投递 →