Principal AI Compiler Engineer

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. The role involves architecting a production‑class AI compiler toolchain, rethinking how AI automates, optimizes, and accelerates every step of building and deploying neural networks on NXP’s SoCs, and working shoulder‑to‑shoulder with visionary engineers to enable adaptive compilers that learn, evolve, and redefine what’s possible for embedded intelligence.

• 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, Tensor Flow, 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.

• 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 and experience with agile development in Scrum (Product Owner or Scrum Master).

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.