Director Machine Learning Engineering - AI/ML Model Compiler and Applications

Position Overview

WHAT YOU DO AT AMD CHANGES EVERYTHING
At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career.

Location: San Jose, CA preferred, Austin, TX may be considered.

The Role

We are looking for a Director, Machine Learning to join our AI/ML Model Compiler and Applications team. If you are excited by the challenge of distributed training of large models on Neural Processing Unit (NPUs), and if you are passionate about improving training efficiency while innovating and generating new ideas, then this role is for you. You will be part of a world-class team focused on addressing the challenge of training generative AI.

Key Responsibilities

• Lead a growing team focused on research, design, and implementation of novel methods for efficient Generative AI.
• Develop ML compilers for low power inference architecture covering all 3 stages – front end, middle end, and backend.
• Define optimized tiling and fusion algorithms across different memory hierarchies.
• Influence the direction of the AMD AI platform.
• Collaborate across teams and stakeholders, including software and hardware groups, to co-optimize model performance.
• Propose and apply innovative model quantization, sparsity, and acceleration algorithms on various AMD platforms (GPU/CPU/AIE).
• Work with open-source frameworks and communities (e.g., PyTorch, Huggingface) to integrate AMD optimized models and libraries.

Required Qualifications

• Experience delivering ML compiler solutions and a deep understanding of NPU hardware architecture.
• Ability to define scalable algorithms for efficient problem mapping.
• Proven experience interfacing with customers to understand requirements and leading a team to develop solutions.
• Academic Credentials: A PhD or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, or a related field.

Preferred Qualifications

• Experience with ML frameworks such as Onnx, PyTorch, or TensorFlow.
• Familiarity with distributed training and frameworks like DeepSpeed.
• Experience with large models in computer vision or language (LLMs) is a plus.
• Excellent C++ and Python programming skills, including debugging, profiling, and performance analysis.
• Experience with ML pipelines.
• Strong communication, leadership, and problem-solving skills.

Benefits & Perks

• Benefits offered: AMD benefits at a glance.