Principal Software Engineer – AI Compiler Engineer (NPU) jobs in United States
cer-icon
Apply on Employer Site
company-logo

AMD · 4 hours ago

Principal Software Engineer – AI Compiler Engineer (NPU)

AMD is a leading company in the computing industry, focused on building innovative products that enhance next-generation computing experiences. The Principal Software Engineer will provide technical leadership for AMD’s NPU compiler stack, focusing on compiler design, optimization strategies, and collaboration with hardware and ML framework teams.

AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
check
Growth Opportunities
badNo H1Bnote

Responsibilities

Provide technical leadership for the architecture and development of AMD’s NPU compiler stack
Design advanced IR transformations, graph‑level optimizations, and hardware‑aware fusion passes
Guide system-wide performance strategies using deep knowledge of HW architecture constraints (memory, parallelism, dataflows)
Collaborate with hardware architects, runtime teams, and ML framework teams (ONNX, PyTorch) to drive co‑design
Drive performance investigations, benchmarking, and software‑hardware co‑optimization
Represent AMD externally through technical publications, open‑source contributions, or industry engagement
Mentor MTS/SMTS engineers and set technical direction for large compiler initiatives

Qualification

C++LLVMMLIRPerformance engineeringPythonCompiler designHardware optimizationAI/ML model executionMentoringTechnical leadershipCollaboration

Required

12+ years of experience in compilers, performance engineering, or system-level software
Strong C++ expertise and proficiency in Python
Extensive hands-on experience with LLVM and MLIR
Experience designing optimizations tightly coupled to hardware architectures (e.g., tensor accelerators, GPUs, NPUs)
Deep understanding of compiler internals: IR design, scheduling, memory optimizations, multi-stage lowering
Demonstrated experience delivering production-quality compiler or systems software
Experience driving technical strategy across teams and influencing product roadmaps
B.S., M.S., or Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, or related field

Preferred

Contributions to open‑source compiler/ML communities (LLVM, MLIR, XLA, TVM, Triton)
Experience with AI/ML model execution, graph optimization, and performance tuning
Experience with domain-specific languages or IRs for machine learning
Knowledge of NPU/GPU execution models, memory hierarchies, and parallelism strategies

Benefits

AMD benefits at a glance.

Company

Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions.

Funding

Current Stage
Public Company
Total Funding
unknown
Key Investors
OpenAIDaniel Loeb
2025-10-06Post Ipo Equity
2023-03-02Post Ipo Equity
2021-06-29Post Ipo Equity

Leadership Team

leader-logo
Lisa Su
Chair & CEO
linkedin
leader-logo
Mark Papermaster
CTO and EVP
linkedin
Company data provided by crunchbase