Sr. Staff Software Engineer - NPU Compiler Solutions jobs in United States
cer-icon
Apply on Employer Site
company-logo

AMD · 1 day ago

Sr. Staff Software Engineer - NPU Compiler Solutions

positionSan Jose, CA
timeFull-time
remoteOnsite
senioritySenior Level, Lead/Staff
money$192K/yr - $288K/yr

AMD is a leading company focused on building innovative products that enhance next-generation computing experiences. They are seeking a Senior Staff Software Engineer to contribute to the architecture and design of the MLIR-centric NPU compiler platform, develop solutions for deploying algorithms on spatial compute devices, and collaborate with cross-functional teams to solve complex problems.

AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
check
Growth OpportunitiesbadNo H1Bnote

Responsibilities

Contributing to the architecture and design of the MLIR-centric NPU compiler platform
Develop and integrate solutions to efficiently deploy algorithms on spatial compute devices, such as NPUs
Work with cross functional teams to identify problems and create solutions
Work with management team on project planning activities

Qualification

C/C++ programmingPythonMLIR experienceAccelerator compiler experienceAuto-optimizationHeterogeneous runtime modelsLinux kernel developmentSoft skills

Required

Strong C/C++ systems programming (modern C++), Python for tooling/prototyping
Hands-on MLIR experience: Dialect design (ops/types/attributes/interface), Pass pipelines, (pattern rewrites, canonicalization/legalization). Proficiency with core dialects: Linalg/Tensor, Affine, SCF, Vector
Accelerator compiler experience: GPU/NPU/AI engines or similar spatial devices; memory hierarchies, streams, scratchpads, NoC-aware optimization
Auto-optimization: practical experience building or integrating auto-tiling, auto-scheduling, and auto-tuning systems
Operator fusion and graph-level optimization for DSP&ML workloads (CNNs, transformers, FFT), layout/dtype transforms, quantization-aware workflows
Solid understanding of heterogeneous runtime models, concurrency, synchronization, and performance profiling
Master's or PhD in Electrical Engineering, Computer Science, or related field

Preferred

Polyhedral/affine analysis, dependence analysis, bank/conflict modeling, memory placement strategies
Open-source contributions to LLVM/MLIR or peer-reviewed publications in compilers/ML systems
ML compiler stacks: IREE, TVM, XLA, Glow, Halide; auto-schedulers/tuners integration
Expertise in Linux kernel/driver development for multi-processor heterogeneous systems
Knowledge of Acceleration platforms like GPU, TPU, APU, FPGAs

Benefits

AMD benefits at a glance.

Company

AMD

twittertwittertwitter
Glassdoor4.1
company-logo
Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions.
dateFounded in 1969
location
Santa Clara, California, USA
people-size10001+ employees
web-link
http://www.amd.com

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

Recent News

Italian Startups - Startupbusiness.it
GENE.01, the first humanoid robot from Generative Bionics
2026-01-07
Mashable
CES 2026: AMD says You aint seen nothing yet on AI
2026-01-07
PCMag Australia
AMD’s New Ryzen AI 400 CPUs Will Put More AI Muscle Into Laptops, Mini Desktops
2026-01-07
Company data provided by crunchbase