在 AMD 的工作改变一切
我们非常关注通过 AMD 技术改变生活,以丰富我们的行业、社区和世界。我们的使命是构建伟大的产品,以加速下一代计算体验——数据中心、人工智能、个人电脑、游戏和嵌入式的基础。在我们使命的背后是 AMD 文化。我们推动创新的极限,以解决世界上最重要的挑战。我们追求卓越的执行,同时保持直接、谦逊、合作,并包容多样化的观点。
AMD 一起进步_
关于我们:
Wave 是一种编程语言,旨在重新定义我们在 AMD GPU 上编程 AI 内核和工作负载的方式。采用独特的子组编程范式,强力支持使用符号表达式推理内核行为,Wave 使得获得高性能并迅速实现变得简单。加入我们,共同开创机器学习加速的未来。
职位概述:
我们正在寻找一位积极进取且经验丰富的编译器和 GPU 性能工程师,在 Wave 的开发中发挥基础架构角色。理想的候选人将具备深厚的编译器设计、优化技术和 GPU 硬件的理解,并对推动人工智能领域高性能计算的边界充满热情。
职责:
设计和实现高级编译器优化过程
开发和优化 GPU 内核
定义并贡献语言特性的设计
将 Wave 与 PyTorch 生态系统无缝集成
开发和维护性能分析工具和方法,以识别和解决性能瓶颈。
紧跟编译器技术、GPU 架构和机器学习硬件的最新进展。
要求:
对机器学习领域的编译器优化和 GPU 性能有强烈的热情。全面理解编译器设计及其与 GPU 硬件的互动。熟练掌握并行计算语言,如 HIP、CUDA、OpenCL 或 Vulkan。具备与 NVIDIA 和 AMD 等领先供应商的 GPU 架构的实际经验。具备性能分析和分析工具的经验。优秀的问题解决和分析能力。强大的沟通和合作能力。
优先考虑:
在生产质量编译器项目中有显著贡献的文档历史。
对高级编译器优化技术有深入理解,尤其是与并行计算相关的技术。
熟悉编译器中间表示,有 MLIR 经验者优先。
在高性能 GPU 应用程序的开发和优化方面有良好的记录。
具备使用 PyTorch 和 TensorFlow 等领先机器学习框架的经验。熟悉低级性能调优及硬件架构对软件性能的影响。
优先学术背景:
通常期望拥有计算机科学、工程或密切相关学科的博士或硕士学位,重点关注编译器设计、并行计算或高性能计算。具有卓越行业经验的候选人也将考虑替代高级学位。
#LI-G11
#LI-HYBRID
提供的福利描述:AMD 福利一览。
AMD 不接受猎头、招聘机构或收费招聘服务的主动简历。AMD 及其子公司是平等机会、包容性的雇主,将考虑所有申请者,而不考虑年龄、血统、肤色、婚姻状况、健康状况、心理或身体残疾、国籍、种族、宗教、政治和/或第三方关系、性别、怀孕、性取向、性别身份、军事或退役军人身份或法律保护的任何其他特征。我们鼓励所有合格候选人申请,并将在招聘和选拔过程的各个阶段根据相关法律满足申请者的需求。