Staff/Sr. Staff Machine Learning Engineer - Model Optimization

Qualcomm • San Diego, CA

98d • $158,400 - $237,600

About The Position

Join the Qualcomm AI Hub team and help developers integrate machine learning into their products and experiences. For this role, we are seeking a talented and motivated Engineer with expertise in the Android platform and on-device AI frameworks. The candidate will have opportunities to develop and support cutting-edge on-device machine learning applications, pushing the boundary of what is possible on mobile and edge hardware. Applications range from latency and power critical streaming video to memory-challenging large language models and Gen AI. Software development would encompass integrating on-device AI frameworks (Qualcomm AI stack, TFLite/LiteRT, ONNX runtime), efficiently pre/post process data in GPU shaders, read sensor data, and putting it all together in a compelling user experience. Profiling across CPU/GPU/NPU to understand performance bottlenecks and minimize memory copies. Interact with customers and design partners as needed and provide engineering roadmap to assist in prioritization of deliverables. Beyond Android, target platforms include Windows and Linux.

Requirements

• 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.

Nice To Haves

• 3+ years of industry experience using data structures and algorithms to ship projects at scale.
• 3+ years of experience with development in Python, modern C++, Java, or Kotlin.
• Solid understanding of Operating System Concepts and Concurrency.
• Strong proficiency in using git for SW development.
• Strong written and verbal communication skills.
• Prior experience with on-device AI runtimes like LiteRT/TFLite, ONNX, and Qualcomm AI stack is a big plus.
• Prior experience working on model quantization or model compression is a big plus.
• Profiling and identifying inefficiencies across mixed CPU/GPU/NPU load.
• Familiarity with python for AI/ML related software development.
• Familiarity with ARM processors, ARM Assembly, Mobile SoC architecture - subsystems, hypervisor/virtualization, security, memory, storage, buses, peripherals, power, thermals.
• Debugging skills, working experience in SW bring up and analyzing crash dumps.

Responsibilities

• Leverages advanced Machine Learning knowledge to extend training or runtime frameworks or model efficiency software tools with new features and optimizations.
• Models, architects, and develops advanced machine learning hardware (co-designed with machine learning software) for inference or training solutions.
• Develops optimized software to enable AI models deployed on hardware (e.g., machine learning kernels, compiler tools, or model efficiency tools, etc.) to allow specific hardware features; collaborates with hardware teams for joint design and development.
• Develops and applies machine learning techniques into products and/or AI solutions to enable customers to do the same.
• Develops, adapts, or prototypes novel machine learning solutions aligned with and motivated by proposals or roadmaps for complex products and working features.
• Oversees and conducts experiments to train and evaluate machine learning models and/or software.

Benefits

• $158,400.00 - $237,600.00 salary range.
• Competitive annual discretionary bonus program.
• Opportunity for annual RSU grants.
• Highly competitive benefits package designed to support success at work, at home, and at play.

What This Job Offers

• Industry: Computer and Electronic Product Manufacturing