Principal AI Compiler Engineer

NXP Semiconductors NV; Austin (TX); On-site; USD 180,000 - 240,000; Full time

Job summary

NXP Semiconductors NV in the United States is seeking an AI Compiler Engineer who thrives at the convergence of AI, compiler tech, and hardware design. You’ll architect and scale a production‑class AI compiler toolchain to map neural networks onto NXP’s SoCs, and work with AI agents to enable adaptive compilers.

You’ll push graph transformations, integration with PyTorch/TF/ONNX, and AI‑guided optimizations across CPUs and accelerators, delivering fast, energy‑efficient results for embedded AI

Qualifications

• MS/PhD (or equivalent experience) in Computer Science, EE, or related field.
• Deep experience building AI compilers, accelerator backends, or graph optimization frameworks.
• Strong expertise in graph optimization and performance optimization for NPUs or custom accelerators.
• Experience with MLIR, LLVM, TVM-like systems, or proprietary compiler IRs.
• Excellent C/C++ and Python skills.
• Solid understanding of AI inference workloads (CNNs, transformers, etc.).
• Strong communication skills and agile development experience.

Responsibilities

• Own the design, implementation, and evolution of an AI compiler toolchain that leverages AI agents to map neural networks onto NXP’s SoC platforms.
• Pioneer graph transformations, lowering, scheduling, and codegen strategies for CPUs and accelerators, driven by AI-powered analytics.
• Build deep integrations with leading AI frameworks (PyTorch, TensorFlow, ONNX) and AI agents for rapid model onboarding.
• Advance quantization, operator fusion, memory planning, and layout transformations with AI-guided design.
• Collaborate with hardware and software architects to co-design compiler and accelerator features for silicon/code alignment.
• Diagnose and reduce performance bottlenecks with AI-enabled profiling and diagnostics.
• Enhance validation, benchmarking, and regression pipelines with AI agents to ensure robust releases.
• Improve developer experience with better usability, diagnostics, and documentation, aided by AI copilots.

Skills

C/C++

Python

Graph optimization

AI compiler experience

Communication skills

Agile/Scrum

MLIR/LLVM/TVM

Education

MS/PhD in CS/EE or related field

Tools

MLIR/LLVM/TVM

Job description

Locations available: San Diego and San Jose, California or Austin, Texas

NXP is searching for a hands‑on AI Compiler Engineer who thrives at the convergence of cutting‑edge AI, compiler tech, and hardware design. Here, you’ll not only architect and scale a production‑class AI compiler toolchain, but also rethink how AI automates, optimizes, and accelerates every step of building and deploying neural networks on NXP’s SoCs. You’ll work shoulder‑to‑shoulder with visionary engineers—both human and AI—enabling adaptive compilers that learn, evolve, and redefine what’s possible for embedded intelligence. With a relentless focus on hardware‑software co‑design, you’ll collaborate across teams to translate high‑level AI models into blazing‑fast, energy‑efficient executables, unlocking the full potential of our silicon for real‑world impact. Innovation here isn’t a catchphrase—it’s your everyday.

Job Responsibilities

• Own the design, implementation, and evolution of an AI compiler toolchain that leverages AI agents to seamlessly map neural networks onto NXP’s SoC platforms.


• Pioneer new graph transformations, lowering, scheduling, and codegen strategies for CPUs and custom accelerators, driven by insights from AI‑powered analytics.


• Build deep integrations with leading AI frameworks (PyTorch, TensorFlow, ONNX, and more), using AI agents to rapidly onboard new model architectures and ops.


• Push the envelope on quantization, operator fusion, memory planning, and layout transformations—combining human expertise and AI‑guided design for state‑of‑the‑art results.


• Partner with hardware and software architects, kernel hackers, and AI agents to co‑design next‑gen compiler and accelerator features, aligning silicon and code for maximum impact.


• Diagnose and crush performance bottlenecks with AI‑enabled profiling and diagnostics, relentlessly tuning for latency, throughput, and power efficiency.


• Level up validation, benchmarking, and regression pipelines by harnessing AI agents—ensuring compiler correctness and world‑class performance, release after release.


• Uplevel the developer experience by streamlining usability, diagnostics, and documentation—AI agents are your copilots for user support, troubleshooting, and rapid iteration.

Qualifications

• MS/PhD (or equivalent experience) in Computer Science, EE, or related field


• Deep experience building AI compilers, accelerator backends, or graph optimization frameworks.


• Strong expertise in graph optimization and performance optimization for NPUs or custom accelerators


• Experience with MLIR, LLVM, TVM‑like systems, or proprietary compiler IRs.


• Excellent C/C++ and Python skills


• Solid understanding of AI inference workloads (CNNs, transformers, perception or generative models)


• Strong communication skills are required, e.g. agile development experience in Scrum team (Product Owner or Scrum Master)

More information about NXP in the United States...

NXP is an Equal Opportunity/Affirmative Action Employer regardless of age, color, national origin, race, religion, creed, gender, sex, sexual orientation, gender identity and/or expression, marital status, status as a disabled veteran and/or veteran of the Vietnam Era or any other characteristic protected by federal, state or local law. In addition, NXP will provide reasonable accommodations for otherwise qualified disabled individuals.