Senior AI Optimization Research Engineer

NXP Semiconductors; Región Centro; Presencial; MXN 1,038,000 - 1,559,000; Jornada completa

Descripción de la vacante

A leading semiconductor company in Mexico is searching for a highly skilled AI Research Engineer/Scientist to contribute to its Edge AI Optimization program. The role involves bridging research with practical implementation, focusing on CNNs and Generative AI for resource-constrained edge devices. Ideal candidates will hold an MSc or Ph.D. in relevant fields and have proven expertise in AI/ML, programming skills in Python and C/C++, and familiarity with hardware constraints. This role offers a chance to shape efficient on-device AI solutions.

Formación

• Deep understanding of Generative AI (Transformers).
• Proficient in systems coding with a focus on hardware interaction.
• Familiarity with embedded systems constraints.

Responsabilidades

• Survey the latest research on neural network quantization.
• Develop novel ideas for hardware constraints.
• Translate prototypes into robust production code.
• Document algorithmic tradeoffs and mentor engineering teams.
• Contribute to NXP’s intellectual property portfolio.

Conocimientos

AI/ML expertise

CNN architectures

Proficient in Python

Proficient in C/C++

Experience with PyTorch

Experience with TensorFlow

Educación

MSc or Ph.D. in Computer Science, Electrical Engineering, or Mathematics

Herramientas

PyTorch

TensorFlow

ONNX

Descripción del empleo

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.

More information about NXP in Mexico.