Edge AI Model Optimization Research Engineer

We at NXP have an environment that fosters innovation. Our team has technology experts who understand the big picture and mentors who coach passionate professionals to work on the most exciting challenges. We share responsibilities in everything we do, where every point of view is valued. Join us! Job Summary We are searching for a highly skilled AI Research Engineer/Scientist with a deep theoretical background and strong systems engineering skills to contribute to our Edge AI Optimization program, NXP’s initiative towards enabling highly efficient Generative and Agentic AI systems on resource-constrained edge devices. You will work at the forefront of innovation, bridging the gap between research and practice, focusing on CNNs, Large Language Model (LLM) and Vision Language Model (VLM) quantization, bringing advanced GenAI and agentic capabilities to NXP NPUs such as Ara-2, directly supporting the future of on-device multimodal intelligence. If you want to shape the future of efficient on-device GenAI and Agentic AI, this is the place to be. --- Job Responsibilities 1. Research: Actively survey the latest research (NeurIPS, ICLR, CVPR) on neural network quantization. Also complementing this with other compression techniques. 2. Prototyping: Develop novel ideas and adapt state-of-the-art methods to meet NXP’s specific hardware constraints and performance targets. 3. Production Implementation: Translate research prototypes into robust, optimized production code (C++/Python), ensuring strict memory and compute efficiency standards. 4. Systems Integration: Document algorithmic tradeoffs, derive deployment recipes, and mentor the engineering team on numerical methods and optimization. 5. IP Generation: Contribute to NXP’s intellectual property portfolio through patents and technical publications. --- Job Qualifications Required Background · Education: MSc or Ph.D. in Computer Science, Electrical Engineering, or Mathematics with a specialization in Machine Learning or Deep Learning. · AI Expertise: Proven experience in AI/ML with a deep understanding of CNN architectures and Generative AI (Transformers). · Technical Stack: Strong hands-on experience with PyTorch, TensorFlow, ONNX, and model conversion/optimization pipelines. · Systems Coding: Proficient in Python and C/C++ with an understanding of how code interacts with underlying hardware. · Embedded Mindset: Familiarity with the constraints of embedded systems (latency, power, memory bandwidth). Preferred · Hardware Acceleration: Experience with NPUs, device-level profiling, and diagnosing memory bottlenecks. · Tooling: Familiarity with MLOps (MLFlow, ClearML) and Yocto Project. · Advanced AI: Experience with custom kernel development is a plus. · Compilers: Knowledge of MLIR or TVM is a significant plus. #LI-FCC3 More information about NXP in Mexico... #LI-fcc3 NXP Semiconductors N.V. (NASDAQ: NXPI) enables a smarter, safer, and more sustainable world through innovation. As the world leader in secure connectivity solutions for embedded applications, NXP is pushing boundaries in the automotive, industrial & IoT, mobile, and communication infrastructure markets. For more information, visit www.nxp.com Bright Minds. Bright Futures. We believe that a key component to growing our business is to develop our people. To enable you to grow your career at NXP, we offer online and offline learning opportunities to help you develop some of your core and professional skills. Commitment At NXP. We recognize NXP is a powerful change agent as we continue to deliver innovative solutions that advance a more sustainable future. We remain steadfast in our commitment to sustainability and making measurable year-on-year progress. Also, we aim to create an inclusive work environment and we will not tolerate racism, discrimination or harassment of any kind. We have programs in place focused on diversity, inclusion and equality. Thank you for considering a career at NXP. To help you prepare for the different steps in our hiring process, see the following useful advice and tips. Are you already an NXP employee? Do not apply here. Instead, you must apply via our internal career page. Thank you for your interest in supporting our recruitment efforts. Please note that NXP operates under a strict Preferred Supplier List (PSL) for all recruitment activities. Any candidate profiles or resume submitted without a prior written agreement or explicit request from our Talent Acquisition team will be considered unsolicited. Such submissions will be deemed free of any obligations, and no fees will be paid by NXP or any of its affiliates, subsidiaries, or divisions - regardless of whether the candidate is hired, either coincidentally or otherwise. Thank you for your understanding.