Ivyea Jobs 5948 roles · 69 companies · 刚刚
摩尔线程 · AI 芯片

GPU计算驱动工程师

北京 校招 · 全职 软件类 AI Infra / 训练系统

职位描述

工作职责: 1. 基于通用计算发展趋势,定义摩尔线程计算编程模型及接口(MUSA),设计和开发相应驱动及运行时程序; 2. 与图形驱动团队协作开发底层公共层; 3. 与开发者工具链及编译器等团队协作开发性能分析工具、调试器、Sanitizer工具、代码自动移植工具等; 4. 为新引入的IP设计开发驱动程序; 5. 协同定位解决上层应用运行时问题,如大规模分布式训练、推理、超算系统的问题; 6. 协同架构、编译器、加速库及开发者工具等团队,联合优化各业务场景端到端性能; 7. 深度参与下一代芯片架构软硬协同设计及优化、性能仿真;

任职要求

任职要求: 1. 精通C/C++编程,具有良好的编程习惯,熟悉常见数据结构和算法,熟悉Linux/Windows系统编程接口; 2. 熟悉计算机体系结构,操作系统原理,编译原理; 3. 熟悉异构计算体系结构,熟悉以下至少一种异构计算API:CUDA、HIP、HSA、OpenCL、Vulkan; 4. 具有AI训练、RL、推理场景硬件协同优化、框架及驱动联合优化经验者优先; 5. 具有较强的定位和解决问题的能力; 6. 具有驱动开发经验、shader编程和优化经验者优先; 加分项: 1. 具有AI应用开发、科学计算应用开发、驱动开发经验、GPU应用开发经验,了解GPU架构; 2. 在国际顶尖会议或期刊(CVPR、ICCV、ECCV、3DV、SIGGRAPH、TOG、TPAMI、ACL、NAACL、EMNLP、AAAI等)上发表过学术论文; 3. 在有影响的计算机科学相关的竞赛中获奖,如ASC、ACM、NOI等;