Principal AI Compiler Engineer

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

NXP is searching for a hands-onAI Compiler Engineerwho thrives at the convergence of cutting-edge AI compiler tech and hardware design. Here youll 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 NXPs SoCs. Youll work shoulder-to-shoulder with visionary engineersboth human and AIenabling adaptive compilers that learn evolve and redefine whats possible for embedded intelligence. With a relentless focus on hardware-software co-design youll 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 isnt a catchphraseits 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 NXPs 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 transformationscombining 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 agentsensuring compiler correctness and world-class performance release after release.
• Uplevel the developer experience by streamlining usability diagnostics and documentationAI 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 addition NXP will provide reasonable accommodations for otherwise qualified disabled individuals.