Software Development Eng.

About the Role

Key Responsibilities

• Design and implement advanced compiler optimization passes
• Develop and optimize GPU kernels
• Define and contribute to the design of language features
• Integrate Wave seamlessly with the PyTorch ecosystem
• Develop and maintain performance analysis tools and methodologies to identify and address performance bottlenecks
• Stay abreast of the latest advancements in compiler technology, GPU architectures, and machine learning hardware

Requirements

• A strong passion for compiler optimization and GPU performance within the domain of machine learning
• A comprehensive understanding of compiler design and its interaction with GPU hardware
• Proficiency in parallel computing languages such as HIP, CUDA, OpenCL, or Vulkan
• Practical experience with GPU architectures from NVIDIA and AMD
• Experience with performance analysis and profiling tools
• Excellent problem-solving and analytical skills
• Strong communication and collaboration abilities

Nice to Have

• A documented history of significant contributions to production-quality compiler projects
• A robust understanding of advanced compiler optimization techniques, particularly those relevant to parallel computing
• Familiarity with compiler intermediate representations, with experience in MLIR being highly advantageous
• A proven track record in the development and optimization of high-performance GPU applications
• Experience with machine learning frameworks such as PyTorch and TensorFlow
• Familiarity with low-level performance tuning and hardware architecture impact on software performance

Qualifications

• A PhD or Master's degree in Computer Science, Engineering, or a closely related discipline with a focus on compiler design, parallel computing, or high-performance computing
• Candidates with exceptional industry experience in lieu of an advanced degree will also be considered

Benefits & Perks

• AMD benefits at a glance