Machine Learning Staff Software Engineer, Pixel Camera

Google; Mountain View (CA); On-site; USD 207,000 - 300,000; Full time

Job summary

Google is seeking a senior software engineer to advance on-device machine learning for mobile and embedded devices. You will own end-to-end model development, from research prototypes to production deployments across hundreds of millions of devices, integrating training pipelines with C++ and the camera stack.

The role focuses on optimizing models for on-device accelerators, quantization, and hardware-aware design, while collaborating with Tensor teams to deliver scalable ML systems.

Qualifications

• Bachelor's degree or equivalent practical experience.
• 8 years of experience with software design and architecture.
• Experience with C, C++, machine learning, and embedded systems.
• Experience with machine learning algorithms.
• Experience with machine learning architecture.
• Experience with machine learning research.

Responsibilities

• Design and implement the real-time on-device machine learning foundation that turns raw camera and sensor input into structured signals driving immediate application behavior.
• Drive the optimization of outlook models for NPU, GPU, and DSP execution, including quantization, hardware-aware architecture design, and custom kernel development, in partnership with the Tensor silicon and compiler teams.
• Own models end-to-end, from prototype through deployment on hundreds of millions of devices - training pipelines in JAX and TensorFlow, C++ integration into the camera pipeline, and long-term maintainability.
• Partner with product, UX, software, and hardware teams to define the requirements for next-generation interactive features, and translate roadmap goals into designs achievable within device constraints.
• Establish best practices for machine learning development, deployment, and evaluation. Define how model quality is measured; and contribute to the long-term technology roadmap.

Skills

Software design

Architecture

Machine learning

Embedded systems

Education

Bachelor’s degree or equivalent

Master’s or PhD in Engineering/CS

Tools

C

C++

JAX/TensorFlow

Job description

In most instances, this position requires in-person interviews as part of the hiring process.

Minimum qualifications

• Bachelor's degree or equivalent practical experience.
• 8 years of experience with software design and architecture.
• Experience with C, C++, machine learning, and embedded systems.
• Experience with machine learning algorithms.
• Experience with machine learning architecture.
• Experience with machine learning research.

Preferred qualifications

• Master's degree or PhD in Engineering, Computer Science, or a related technical field.
• Experience taking machine learning models from research prototype to production on mobile or embedded devices, and owning them after launch.
• Experience optimizing models for on-device accelerators through quantization, hardware-aware architecture design, or custom kernels.
• Experience building and training models in JAX/TensorFlow.

About The Job

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward.

The Google Pixel team focuses on designing and delivering the world's most helpful mobile experience. The team works on shaping the future of Pixel devices and services through some of the most advanced designs, techniques, products, and experiences in consumer electronics. This includes bringing together the best of Google's artificial intelligence, software, and hardware to build global smartphones and create transformative experiences for users across the world.

Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $207000 - $300000 (USD) + 20% bonus target + equity + benefits

Responsibilities

Learn more about benefits at Google .

• Design and implement the real-time on-device machine learning foundation that turns raw camera and sensor input into structured signals driving immediate application behavior.
• Drive the optimization of outlook models for NPU, GPU, and DSP execution, including quantization, hardware-aware architecture design, and custom kernel development, in partnership with the Tensor silicon and compiler teams.
• Own models end-to-end, from prototype through deployment on hundreds of millions of devices - training pipelines in JAX and TensorFlow, C++ integration into the camera pipeline, and long-term maintainability.
• Partner with product, UX, software, and hardware teams to define the requirements for next-generation interactive features, and translate roadmap goals into designs achievable within device constraints.
• Establish best practices for machine learning development, deployment, and evaluation. Define how model quality is measured; and contribute to the long-term technology roadmap.

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form .