Back to jobs
Google

Staff Software Engineer, On-Device Machine Learning

Posted 1 weeks ago

Job Description

  • Create roadmaps for developer-facing APIs, SDKs, and tools, ensuring they meet the evolving needs of LLM workflows.
  • Solve technically challenging problems that exceed the scope of a generalist software engineer, specifically around optimizing GenAI performance across heterogeneous hardware (CPUs, GPUs, and EdgeTPUs).
  • Guide the team in designing resilient systems, anticipating scaling bottlenecks or shifts in usage as LLMs become increasingly complex.
  • Coordinate efforts across multiple groups, including Android ML, ML Compiler, and DeepMind, to co-design performance and evaluation workflows.
  • Provide technical mentorship and implement new practices that address team needs and increase the velocity of your teammates.

See Your Match Score

Sign up and Renata will show you how this job matches your skills and experience.

Staff Software Engineer, On-Device Machine Learning at Google | Renata