Staff Software Engineer, On-device Machine Learning

Google Google · Big Tech · Sunnyvale, CA +1

Staff Software Engineer role focused on leading the technical direction of On-Device Machine Learning (ODML) and generative AI (GenAI) at Google. The role involves creating roadmaps for developer-facing APIs, SDKs, and tools, solving complex optimization problems for GenAI performance on mobile hardware, designing resilient systems, and coordinating efforts across multiple teams. The position requires extensive experience in software development, ML infrastructure, and launching software products, with a focus on optimizing LLM workflows for on-device deployment.

What you'd actually do

  1. Create roadmaps for developer-facing APIs, SDKs, and tools, ensuring they meet the evolving needs of LLM workflows.
  2. 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).
  3. Guide the team in designing resilient systems, anticipating scaling bottlenecks or shifts in usage as LLMs become increasingly complex.
  4. Coordinate efforts across multiple groups, including Android ML, ML Compiler, and DeepMind, to co-design performance and evaluation workflows.
  5. Provide technical mentorship and implement new practices that address team needs and increase the velocity of your teammates.

Skills

Required

  • Bachelor’s degree or equivalent practical experience.
  • 8 years of experience in software development.
  • 5 years of experience testing and launching software products, and 3 years of experience with software design and architecture.
  • 5 years of experience with one or more of the following: speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field.
  • 5 years of experience with ML design and ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning).

Nice to have

  • Master’s degree or PhD in Engineering, Computer Science, or a related technical field.
  • 8 years of experience with data structures and algorithms.
  • 3 years of experience in a technical leadership role leading project teams and setting technical direction.
  • 3 years of experience working in a complex, matrixed organization involving cross-functional or cross-business projects.
  • Experience in GenAI end-to-end workflows.
  • Experience with mobile app development experience and familiarity with iOS/Android build/testing systems.

What the JD emphasized

  • On-Device Machine Learning
  • generative AI (GenAI)
  • Large Language Models (LLMs)
  • mobile hardware
  • developer tools
  • optimizing GenAI performance
  • heterogeneous hardware
  • ML Compiler
  • DeepMind
  • performance and evaluation workflows
  • ML infrastructure
  • model deployment
  • model evaluation
  • fine tuning

Other signals

  • On-Device Machine Learning
  • generative AI (GenAI)
  • Large Language Models (LLMs)
  • mobile hardware
  • developer tools
  • optimizing GenAI performance
  • heterogeneous hardware
  • ML Compiler
  • DeepMind
  • performance and evaluation workflows
  • ML infrastructure
  • model deployment
  • model evaluation
  • fine tuning