Qualcomm

Staff Machine Learning Engineer - Model Optimization & Quantization

Qualcomm$160K — $240K *
Consumer Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's/Master's/PhD in Computer Science, Engineering, or related field with relevant engineering experience.
  • 3+ years in machine learning, deep learning, or AI infrastructure (preferred).
  • Proficient in Python with experience in PyTorch, ONNX, and/or TensorFlow.
  • Solid understanding of various neural network architectures (CNNs, Transformers, LLMs).
  • Hands-on experience with model quantization techniques.

Responsibilities

  • Design and maintain quantization algorithms within the AIMET framework.
  • Implement advanced quantization strategies for LLMs and generative AI.
  • Build tooling for profiling model accuracy degradation due to quantization.
  • Integrate AIMET workflows with popular ML frameworks.
  • Develop user-friendly APIs and tooling for external developers.
  • Quantize and validate diverse model families for AI Hub deployment.
  • Manage end-to-end model optimization to meet performance targets.

Benefits

  • Competitive annual discretionary bonus program.
  • Opportunity for annual RSU grants.
  • Comprehensive benefits package supporting work-life balance.
Full Job Description
Company:
Qualcomm Technologies, Inc.

Job Area:
Engineering Group, Engineering Group > Machine Learning Engineering

General Summary:

About the Role

Join the Qualcomm AI Hub team and help developers integrate machine learning into their products and experiences: https://aihub.qualcomm.com/.

In this role you will develop tools to help developers optimize and deploy machine learning models on edge and mobile hardware. AIMET is Qualcomm's open-source library for state-of-the-art model quantization, and compression techniques. You will develop and support cutting-edge model optimization workflows - pushing the boundary of what's possible on resource-constrained hardware. Applications range from quantizing large language models (LLMs) and generative AI models to compressing latency-critical vision, audio, and multimodal networks for deployment on Qualcomm Snapdragon and other edge SoCs.

For this role we are seeking a talented and motivated Staff Software Engineer with expertise in the optimizing and deploying ML models - especially for edge devices.

What You'll Do

  • Design, develop, and maintain quantization algorithms and compression pipelines within the AIMET framework (PTQ, QAT, mixed-precision, AdaScale etc.)
  • Implement advanced quantization techniques including weight-only quantization, activation quantization, KV-cache quantization, and sub-4-bit quantization for LLMs and generative AI models
  • Build tooling to analyze, profile, and debug model accuracy degradation caused by quantization
  • Integrate AIMET workflows with popular ML frameworks - PyTorch and ONNX
  • Develop APIs and developer-facing tooling to make AIMET accessible and easy to use for external customers and design partners
  • Integrate AIMET in AI Hub Workbench Quantize job to enable Quantization at large scale.
  • Own end-to-end quantization and optimization of models published on Qualcomm AI Hub, ensuring they meet accuracy, latency, and power targets on Qualcomm hardware
  • Quantize and validate a broad range of model families - vision transformers, LLMs, diffusion models, speech, and multimodal architectures - for deployment via AI Hub
  • Develop and maintain automated quantization pipelines and evaluation harnesses to scale model onboarding across AI Hub's growing model catalog


Minimum Qualifications:
• Bachelor's degree in Computer Science, Engineering, Information Systems, or related field and 4+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.
OR
Master's degree in Computer Science, Engineering, Information Systems, or related field and 3+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.
OR
PhD in Computer Science, Engineering, Information Systems, or related field and 2+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.

Preferred Qualifications:

  • 3+ years of industry experience in machine learning, deep learning, or AI infrastructure
  • Strong proficiency in Python, with hands-on experience in PyTorch, ONNX and/or TensorFlow
  • Solid understanding of neural network architectures - CNNs, Transformers, LLMs, diffusion models, multimodal models
  • Experience with model quantization techniques - PTQ, QAT, weight-only quantization, mixed-precision, sub-4-bit methods
  • Hands-on experience quantizing LLMs (GPT, LLaMA, Mistral, Falcon, or similar families) for inference optimization
  • Familiarity with AIMET, GPTQ, AWQ, SmoothQuant, or similar quantization frameworks is a strong plus
  • Experience working with ONNX, TFLite/LiteRT, or other model interchange formats
  • Understanding of hardware constraints: memory bandwidth, compute precision (INT4/INT8/FP16/BF16), and NPU/DSP execution
  • Experience collaborating across teams or BUs to drive technical alignment and model delivery
  • Proficiency with git and software development best practices
  • Strong written and verbal communication skills - ability to write clean APIs, documentation, and engage directly with external developers
  • Experience with C++ for performance-critical components is a bonus
  • Familiarity with ARM processors and mobile SoC architecture (Snapdragon) is a plus
  • Experience with automated evaluation pipelines and model benchmarking at scale is a plus


Level of Responsibility

  • Works independently with minimal supervision
  • Provides technical guidance and mentorship to other team members
  • Decision-making is significant and affects work beyond the immediate team
  • Requires strong communication skills to convey complex quantization concepts to varied audiences - from hardware engineers and BU partners to external researchers and application developers
  • Has meaningful influence on the AIMET product roadmap, AI Hub model catalog, and cross-BU quantization strategy
  • Tasks are open-ended; planning, prioritization, and problem-solving are core to the role


Pay range and Other Compensation & Benefits:
$160,500.00 - $240,700.00

The above pay scale reflects the broad, minimum to maximum, pay scale for this job code for the location for which it has been posted. Even more importantly, please note that salary is only one component of total compensation at Qualcomm. We also offer a competitive annual discretionary bonus program and opportunity for annual RSU grants (employees on sales-incentive plans are not eligible for our annual bonus). In addition, our highly competitive benefits package is designed to support your success at work, at home, and at play. Your recruiter will be happy to discuss all that Qualcomm has to offer - and you can review more details about our US benefits at this link.

If you would like more information about this role, please contact Qualcomm Careers.

About Qualcomm

Qualcomm Ventures is the investment arm of Qualcomm Incorporated. Founded in 2000, Qualcomm Ventures is a corporate venture capital fund with over 150 active portfolio companies and more than 20 exits over a billion dollars, including 99 Taxis, Cruise Automation, Fitbit, Invensense, NQ Mobile, Waze, and more. As a global investor, Qualcomm Ventures helps connect entrepreneurs to the resources, relationships, and deep industry expertise they need to succeed in the mobile technology ecosystem.

Qualcomm Careers

Joining Qualcomm offers more than just a job opportunity; it's a gateway to a career infused with innovation, leadership, and growth. As a pivotal leader in the world of wireless technology, Qualcomm stands at the forefront of digital communication advancements. Our team of professionals is dedicated to pushing the boundaries of what's possible, making this an ideal time to become part of our global community.

Work You’ll Do

At Qualcomm, you will collaborate with some of the brightest minds in the industry, engaging in work that transforms the way the world connects, computes, and communicates. Our diverse team is driven by a shared passion for creating path-breaking wireless technologies that empower mobile ecosystems worldwide.

Innovate and Grow

Embrace the opportunity to innovate alongside leaders in the field and contribute to projects that have a global impact. Qualcomm is committed to fostering a culture of innovation and continuous improvement, ensuring that every team member has the opportunity to make a significant impact.

Professional Growth and Development

Qualcomm is dedicated to the professional growth of its employees, offering unparalleled benefits, diverse career paths, and extensive training programs that encourage professional and personal development. Whether you're looking for leadership roles or specialized technical positions, Qualcomm provides the resources and support to help you drive your career forward.

Diversity and Inclusion

We believe that a diverse workforce fuels our innovation and reflects our commitment to making a positive impact. Qualcomm’s inclusive culture and diversity training programs are designed to promote an environment where all employees can thrive.

Internship Programs

Start your career with Qualcomm through our dynamic internship programs. These opportunities allow you to apply your skills in real-world scenarios, providing a robust foundation for future employment. Internships at Qualcomm are characterized by meaningful projects and the chance to network with industry leaders.

Join Our Team

Explore the numerous job opportunities at Qualcomm, from engineering to marketing, and discover how your skills and interests align with our mission. We are continuously hiring creative and driven individuals who are ready to contribute to our culture of innovation.

Prepare for Your Interview

Ready to join our team? Prepare your resume to highlight your relevant experience and skills. Our interview process is designed to understand your capabilities and how they align with our goals at Qualcomm. We look for passionate, curious, and innovative team players who are ready to take the next step in their careers.

Stay Connected

Keep up to date with the latest at Qualcomm by following our careers blog. Gain insider perspectives and industry-leading insights that can help you navigate your professional journey.

Career Opportunities Await

At Qualcomm, your career is what you make of it. With support for your ambitions and a network of global professionals, the opportunities to advance and excel are nearly limitless. Join us and be part of a team that’s leading the world in next-generation technology.

Search Qualcomm Jobs

Discover the positions that match your skills and interests. We are looking for individuals who are ready to make an impact and excel in a fast-paced, innovative environment.

Explore Qualcomm Careers

Whether you're seeking an internship, a first job, or a leadership position, Qualcomm offers a range of opportunities across various disciplines. Let your career journey begin here, where innovation, leadership, and growth come together to create extraordinary outcomes.
Learn more about Qualcomm
Size
45,000 employees
Market Cap
$122.5 billion
Industry
Net Income
$6.7 billion
Founded
1985
5 Year Trend
+14.7%
Revenue
$26.6 billion
NASDAQ

Similar Jobs

More Jobs at Qualcomm

More Consumer Technology Jobs

Find similar Staff Machine Learning Engineer - Model Optimization & Quantization jobs: