Team Lead - AI Compiler Engineer (NPU)

AMD - Advanced Micro Devices

Cologne, Germany

THE ROLE:

We are seeking a highly skilled and motivated AI Compiler Engineer to lead the development of AMD’s next-generation compiler stack for NPUs. This is a hands-on leadership role where you will guide a team of talented engineers, drive architectural decisions, and contribute directly to the implementation of novel compiler optimizations and fusion techniques. You’ll be working at the intersection of AI and systems, helping shape the future of high-performance AI workloads on AMD hardware.

THE PERSON:

You are a technical leader with a passion for compilers, AI, and performance optimization. You thrive in a fast-paced environment, enjoy mentoring others, and are excited by the opportunity to work on cutting-edge technologies. You bring deep technical expertise, a collaborative mindset, and a drive to deliver high-impact results.

KEY RESPONSIBILITIES:

Lead the design and development of the compiler stack for AMD’s NPU architecture;
Mentor and guide junior engineers, fostering a culture of technical excellence;
Architect and implement advanced graph fusion and optimization techniques;
Collaborate with hardware architects, ML framework teams, and runtime engineers;
Drive performance analysis and tuning across ML models and benchmarks;
Contribute to the evolution of MLIR and LLVM-based infrastructure;
Stay current with trends in AI compilers, ML frameworks, and hardware acceleration.

REQUIRED EXPERIENCE:

Strong C++ and Python programming skills;
Deep understanding of compiler internals, IRs, and optimization pipelines;
Hands-on experience with MLIR, LLVM, and ONNX/ONNX Runtime;
Familiarity with PyTorch, Triton, or other ML frameworks;
Proven track record of technical leadership and mentoring;
Experience with performance tuning and low-level system optimization;
Excellent communication and collaboration skills.

PREFERRED QUALIFICATIONS:

Experience with NPU, GPU, or other AI accelerator architectures;
Contributions to open-source compiler or ML projects;
Background in machine learning or numerical computing;
Familiarity with software development tools (e.g., Git, CI/CD, profilers, debuggers).