← WorkMundi · 1M+ jobs from around the world, liveSign inCreate free account

Edge AI Model Optimization Software Engineer (Mumbai)

NXP Semiconductors · Mumbai City

📅 07/08/2026
🔔 Alert me about jobs like this
No password, no sign-up. Just the email — and you can leave the list anytime.
🔓 Apply — free →
Opens this job on WorkMundi. The account is free and takes under a minute.

See the other 113,540 jobs in India →

🎁 Before you apply, rehearse this interview. Create your free WorkMundi account and get an Interview Training on HelpsYouSpeak — no cost, no card. I want my training →
Edge AI Model Optimization Software Engineer Experience: Not Available to Not Available years Location: Guadalajara Skills: Python, C/C++, Machine Learning, Deep Learning, AI, ML, CNN architectures, Generative AI, Transformers, Quantization, PyTorch, ONNX, hardware accelerators, device-level profiling, memory bottlenecks, embedded systems, GPTQ, Smoothquant, MLIR, TVM Job Summary We are looking for a hands-on AI Optimization Engineer who thrives at the intersection of algorithms, systems engineering, and production-grade software. We are looking for someone who wants to work at the heart of state-of-the-art AI, not yesterdays models, but the frontier: Generative AI, Transformers, Vision-Language Models (VLMs), and the emerging wave of Agentic AI systems. In this role, you will contribute to the engineering of the core optimization technology that makes these advanced models run efficiently on NXPs next-generation edge platforms, such as Ara 2. While Neural Network Quantization will be your primary focus, your impact will go far beyond it. You will architect high-performance software and collaborate across compiler, and hardware teams. Your work will directly define how frontier AI models are executed on-device, enabling fast, reliable, and power-efficient GenAI on resource-constrained platforms. If you want to work where AI innovation meets real-world engineering, and build the infrastructure that defines what tomorrows smart, autonomous, AI-powered devices can do, this is the role for you. Job Responsibilities 1. Optimization Tools: Own and evolve our production-grade optimization tools. You will design and implement quantization features, including mixed-precision flows that are robust enough for global deployment. 2. Pipelines: Design and maintain scalable PTQ (Post-Training Quantization) and QAT (Quantization-Aware Training) workflows, ensuring seamless integration with other optimizations and downstream deployment stacks. 3. Applied Innovation & Tooling: Collaborate on ProofofConcepts (POCs) for state-of-the-art quantization techniques. You wont just prototype; you will evaluate the real-world impact of novel ideas and engineer the "bridge" that turns successful experiments into production-ready deployment recipes and developer tooling. 4. Numerical & Hardware Rigor: Apply your math foundation to implement approximation algorithms (range estimation, bias correction, BN-folding) while ensuring bit-exactness on target hardware. You will bridge the gap between abstract math and physical constraints like accumulator widths, saturation, and rounding behaviors. 5. Systems Performance: Profile and optimize the "hot paths" of our optimization toolchain to meet strict memory and compute-constrained targets. 6. Cross-Functional Leadership: Act as the technical bridge between AI Research and Hardware Engineering, providing quantified guidance on how choices impact model accuracy and performance. 7. Deployment Architecture: Document algorithmic tradeoffs and derive "gold-standard" deployment recipes, acting as technical mentor for other engineering teams, ensuring deployment strategies are scalable and repeatable. Job Qualifications Required Background Education: MSc or Ph.D. (is a plus) in Computer Science, Electrical Engineering, or Mathematics with a specialization in Machine Learning or Deep Learning. Systems Programming: Mastery of Python and C/C++. You should be comfortable with memory management and understanding how code maps to hardware (CPUs/NPUs). AI Expertise: Proven experience in AI/ML with a good understanding of CNN architectures and Generative AI (Transformers). AI Optimization: Experience with (or a strong desire to learn) quantization workflows and troubleshooting accuracy regressions. Technical Stack: Strong hands-on experience with PyTorch, ONNX, and other AI/ML frameworks. Preferred Hardware Acceleration: Experience with hardware accelerators, device-level profiling, and diagnosing memory bottlenecks. Embedded Mindset: Familiarity with the constraints of embedded systems (latency, power, memory bandwidth). Advanced AI: Experience implementing state-of-the-art quantization for generative AI (e.g., GPTQ, Smoothquant, etc). Compilers: Knowledge of MLIR or TVM is a significant plus. What You Will Gain Be part of a pioneering team shaping the future of AI and edge computing. Work on innovative projects that solve real-world challenges. Opportunity to grow with a dynamic, forward-thinking company. Competitive salary, benefits, and a collaborative work workplace. #LI-FCC3 More information about NXP in Mexico. #LI-fcc3 .
Read the rest of the job →
For people searching Engineer

144,883 engineer jobs are open right now

Here's how to pick the right one and stand out in your application.

144.883Jobs
31.687IN
81%EN

That number is real. WorkMundi's database shows 144,883 open engineer roles across the world. India has the most with 31,687 jobs, followed by the United States with 30,084. If you just finished reading one job ad and felt paralyzed by choice, you're not alone—but this scale is actually an advantage. It means you can afford to be selective.

Start by geography and language. The majority of engineer ads—117,837 of them—have the job posting text written in English. Use that as one filter, but remember: the ad text language tells you nothing about whether the role actually requires you to speak English day-to-day. Read the job description carefully. Then check which countries have the volume you're targeting. Singapore, Poland, and Australia round out the top five after India and the US.

Next, learn who's hiring. Accenture has posted 2,801 engineer roles. andurilindustries, speechify, and jobgether are also actively recruiting. If you're applying to one of these names, research their hiring patterns and interview style before you apply. That homework pays off.

When you interview, expect the question every engineer hears: 'Tell me about a time you had to debug a problem that wasn't in your job description.' Have a specific story ready—not a general one. Name the tools, the deadline pressure, and what you learned. Hiring managers listen for whether you see problem-solving as part of the role itself, not a favour.

👁 21 have read this
0 comments
Want to comment?

Leave your e-mail to comment, react and follow the posts for your role. It is free.

Similar jobs

Job on WorkMundi — the world's largest job board. See more jobs from every continent, updated live.

📢
🎁

Before you apply, rehearse this interview.

Create your free WorkMundi account and get an Interview Training on HelpsYouSpeak — no cost, no card.

I want my training →