Senior On-Device ML Engineer - Camera

Google; Town of Montana (WI); On-site; USD 207,000 - 300,000; Full time

Benefits offered by this job

Equity

Bonus target

Benefits

Job summary

Google's Pixel Camera team is seeking a Machine Learning Staff Software Engineer to design and implement the real-time, on-device machine learning foundation that drives camera features.

You will own ML models from research prototype to production on hundreds of millions of devices, optimizing for NPU, GPU and DSP, and collaborating with Tensor silicon and compiler teams, as well as product, UX, and hardware teams.

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

C

C++

Machine learning

Embedded systems

Software design and architecture

ML algorithms

ML architecture

ML research

Education

Bachelor's degree

Master's degree

PhD

Tools

JAX

TensorFlow

Job description

Google's Pixel Camera team is seeking a Machine Learning Staff Software Engineer to design and implement the real-time, on-device machine learning foundation that drives camera features.

You will own ML models from research prototype to production on hundreds of millions of devices, optimizing for NPU, GPU and DSP, and collaborating with Tensor silicon and compiler teams, as well as product, UX, and hardware teams.