← All roles

Machine Learning Staff Software Engineer, Pixel Camera

Google
Apply on company site
Hybrid (Mountain View, CA)$207k - $300kFull-timePosted 1mo ago

About the role

Senior Machine Learning staff software engineer role focused on on-device ML for Google Pixel Camera, owning end-to-end ML models from prototype to production on mobile devices, with emphasis on real-time on-device ML foundations and hardware acceleration optimization.

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.

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.

Skills

Required:Software design and architectureCC++Machine learningJAX/TensorFlow
Nice to have:On-device ML optimization

Benefits

  • Learn more about benefits at Google.
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Google

Google Search homepage

llmtext-to-speechvoice-ai-stack
Company profileWebsite

More open roles at Google