Qualcomm
Posted 6d ago

Staff Machine Learning Engineer – Model Optimization & Quantization

Qualcomm
Santa Clara, California, United States
$161k-$241k/yrOnsiteFull Time
Responsibilities
  • developing algorithms
  • building tooling
  • optimizing models
Requirements
  • Bachelor's and 4+ years
  • Master's and 3+ years, or PhD and 2+ years in engineering or related work. Requires Python
  • ML frameworks
  • Quantization expertise, and software development experience
Technical tools mentioned
AIMETPyTorchONNXTensorFlowGPTQAWQSmoothQuantTFLite/LiteRTgitC++PythonAI Hub WorkbenchSnapdragon

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 

Qualcomm is an equal opportunity employer. If you are an individual with a disability and need an accommodation during the application/hiring process, rest assured that Qualcomm is committed to providing an accessible process. You may e-mail [email protected] or call Qualcomm's toll-free number found here. Upon request, Qualcomm will provide reasonable accommodations to support individuals with disabilities to be able participate in the hiring process. Qualcomm is also committed to making our workplace accessible for individuals with disabilities. (Keep in mind that this email address is used to provide reasonable accommodations for individuals with disabilities. We will not respond here to requests for updates on applications or resume inquiries).

To all Staffing and Recruiting Agencies: Our Careers Site is only for individuals seeking a job at Qualcomm. Staffing and recruiting agencies and individuals being represented by an agency are not authorized to use this site or to submit profiles, applications or resumes, and any such submissions will be considered unsolicited. Qualcomm does not accept unsolicited resumes or applications from agencies. Please do not forward resumes to our jobs alias, Qualcomm employees or any other company location. Qualcomm is not responsible for any fees related to unsolicited resumes/applications.

EEO Employer: Qualcomm is an equal opportunity employer; all qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or any other protected classification.

Qualcomm expects its employees to abide by all applicable policies and procedures, including but not limited to security and other requirements regarding protection of Company confidential information and other confidential and/or proprietary information, to the extent those requirements are permissible under applicable law.

 

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

Designs and manufactures semiconductors and wireless telecommunications products.

Similar jobs

Machine Learning Engineer roles near Santa Clara, California
1d
Save
Mark Applied
Hide
Founding Engineer - Machine Learning
Mountain View, California, United States
$220k-$300k/yr OnsiteFull Time
Clera
Clera: AI talent agent matching professionals with high-growth startup roles
3+ YOERequires 3–10 years of ML engineering, applied science, or research engineering experience; Python and PyTorch, TensorFlow, or JAX; distributed systems, cloud ML infrastructure, MLOps, and large-scale data experience.
Python, PyTorch, TensorFlow, JAX, AWS, GCP, Azure, Weights & Biases, MLflow
1d
Save
Mark Applied
Hide
Machine Learning Engineer
San Francisco, California, United States
$170k-$300k/yr HybridFull Time
Clay
Clay: Platform for automated lead enrichment and sales workflows.
5+ YOE5+ years in machine learning engineering or ML-heavy software engineering, production ML features, strong coding, LLMs or classical ML, data-intensive systems, and product-oriented problem solving.
LLMs, Snowflake, dbt, Dagster
1d
Save
Mark Applied
Hide
Machine Learning Engineer - Satellite Capacity Optimization & Planning
Carlsbad or San Jose or San Francisco or New York City
$141k-$222k/yr OnsiteFull Time
Viasat
ViasatNASDAQ: VSAT: Provider of global satellite-based connectivity and secure communication solutions.
7+ YOERequires 7+ years in ML or optimization, optimization techniques, production cloud ML systems, ML frameworks, geospatial visualization, containerized development, SQL, RESTful APIs, and up to 10% travel.
AWS, GCP, TensorFlow, PyTorch, SQL, RESTful APIs, Airflow, Athena, BigQuery, ECS, Batch
2d
Save
Mark Applied
Hide
Staff Machine Learning Engineer, Generative AI Modeling and Inference
Los Angeles or Seattle or Palo Alto or New York City or Bellevue
$229k-$343k/yr OnsiteFull Time
Snap
SnapNYSE: SNAP: Provides visual messaging software and augmented reality wearable devices.
8+ YOEBachelor's degree or equivalent experience and 8+ years of post-bachelor's machine learning experience, with expertise in generative modeling, computer vision, deep learning, and ML frameworks.
TensorFlow, PyTorch, JAX, MLX, scikit-learn
2d
Save
Mark Applied
Hide
(USA) Staff, Machine Learning Engineer
Sunnyvale, California, United States
$169k-$338k/yr OnsiteFull Time
Walmart
WalmartNYSE: WMT: Operates a chain of hypermarkets, discount stores, and grocery stores.
4+ YOEBachelor's degree and 4 years, or 6 years of experience, in software engineering, machine learning engineering, AI systems, or a related field. Requires ML, MLOps, cloud, and programming expertise.
TensorFlow, PyTorch, Scikit-learn, Python, SQL, AWS, GCP, Azure, Kubernetes, CI/CD, MLOps, Web Content Accessibility Guidelines (WCAG) 2.2 AA
2d
Save
Mark Applied
Hide
Machine Learning Engineer
Chicago or Washington or Tallahassee or Sarasota or Hartford or Nashville or Costa Mesa or San Jose or Atlanta or Boston or Burlington or Cleveland or Columbus or Dallas or Denver or Fort Lauderdale or Fort Wayne or Grand Rapids or Indianapolis or Knoxville or Lexington or Livingston or Plano or Louisville or Los Angeles or Miami or The Woodlands or New York City or Oakbrook Terrace or Sacramento or San Francisco or South Bend or Springfield or Tampa or Houston or Austin or Charlotte
$62k-$100k/yr OnsiteFull Time
Crowe
Crowe: Global professional services firm providing audit, tax, and consulting.
Expert Python and Git skills; ability to develop, test, debug, and document production AI solutions, navigate ambiguity, communicate technical details, and support ethical and security reviews.
Python, Git, ChatGPT Enterprise, Microsoft Copilot, Azure AI Foundry
2d
Save
Mark Applied
Hide
Lead Machine Learning Engineering, (Hybrid)
Seattle or San Jose
$198k-$250k/yr HybridFull Time
Cisco
CiscoNASDAQ: CSCO: Develops and sells networking hardware and cybersecurity software.
8+ YOESTEM degree and 8+ years relevant experience, or equivalent advanced-degree pathways; 3+ years scaling ML datasets; 5+ years programming and ML framework experience.
Python, C++, Go, PyTorch, TensorFlow, Spark, Ray, Beam, Large Language Models (LLMs)
2d
Save
Mark Applied
Hide
Staff Machine Learning Engineer, Generative AI Modeling and Inference
Los Angeles or Seattle or Palo Alto or New York City or Bellevue
$195k-$343k/yr OnsiteFull Time
Snap
SnapNYSE: SNAP: Develops social media applications and augmented reality technology.
8+ YOEBachelor's degree or equivalent experience and 8+ years of post-bachelor's ML experience, or advanced degree with equivalent experience. Requires computer vision or generative modeling and ML framework experience.
TensorFlow, PyTorch, JAX, MLX, scikit-learn, GPU, CPU, NPU, RSUs