AI Software Engineer

About this role

Job Title:
AI Software Engineer
Company:
NXP Semiconductors México, S. de R.L. de C.V.
NXP offers an inclusive, collaborative, and innovation-driven work culture where engineers have the freedom to explore ideas, contribute to cutting-edge technologies, and grow their technical depth.


Job Summary
As part of ongoing enhancements to our
Discrete NPU Model Zoo
, we are looking for AI Engineers to contribute to the design, development, and optimization of model‑zoo infrastructure supporting next‑generation
computer vision
,
generative AI
, and
LLM workloads
.


You will work on developing compilation, inference, and performance workflows for AI accelerators, while helping build end‑to‑end demos and benchmarking pipelines.


CNN Model Zoo Responsibilities

• Collaborate on scaling up our CNN‑based model zoo.
• Add new models and their variants in both Python and C++.
• Improve accuracy across ~50 existing models.
• Re‑implement variant models in PyTorch for integration with the MCW tool.
• Use the MCW tool for optimization, pruning, and performance tuning.
• Automate model testing, validation, and optimization flows where feasible.
• Develop gstreamer / NXP nnstreamer plugins and end‑to‑end inference pipelines.
• Build high‑quality demos showcasing model capabilities on NPU accelerators.

LLM/VLM Model Zoo Responsibilities

• Contribute to the development and expansion of the LLM model zoo.
• Use Hugging Face APIs to build required pipelines and workflows (Python and C++).
• Integrate and support multiple variants of LLMs and VLMs.
• Develop automated testing frameworks and Jenkins‑based CI pipelines.
• Execute benchmarking, performance profiling, and validation of LLM/VLM models.
• Build demos demonstrating large‑model capabilities on our NPU platforms.

Job Qualifications

• Bachelor’s degree in Electronics/Computer/Systems/Robotics Engineering or similar with 3+ years of experience in Machine Learning or relate...