Staff Software Engineer, On-Device Machine Learning Infrastructure

Google LLC
  • Sunnyvale, CA
    9 days ago

    Job Description

    Staff Software Engineer, On-Device Machine Learning Infrastructure - Google Careers

    Careers

    Careers

    Skip navigation links

    homehome

    Home

    Home

    work_outlinework_outline

    Jobs

    Jobs

    noogler_hatnoogler_hat

    Students

    Students

    googlegoogle

    How we work

    How we work

    handymanhandyman

    How we hire

    How we hire

    person_outlineperson_outline

    Your career

    Your career

    help_outline

    Help link

    feedback

    Send feedback

    more_vert

    • Help
    • Send Feedback

    Sign in

    Careers

    Careers

    homeHome

    work_outlineJobs

    expand_more

    noogler_hatStudents

    expand_more

    googleHow we work

    expand_more

    handymanHow we hire

    expand_more

    person_outlineYour career

    expand_more

    job details

    arrow_back

    Back to jobs search

    Jobs search results

    3,445 jobs matched

    • Senior Software Engineer, Vertex AI, Workbench

    Warsaw, Poland

    • Customer Growth Associate, SBS Scaled, Google Customer Solutions, Early Careers (English, Polish)

    Dublin, Ireland

    • Software Engineer, GDC AI Applications and Agents

    Sunnyvale, CA, USA

    • Software Engineer, PhD, Early Career, 2026

    Bengaluru, Karnataka, India; Hyderabad, Telangana, India

    • Senior Software Engineer, Site Reliability Engineering, Cloud Storage

    Dublin, Ireland

    • Strategy Associate, YouTube

    San Bruno, CA, USA

    • Power Delivery Design Engineer, Google Cloud

    Tel Aviv, Israel; Haifa, Israel

    • Staff Site Reliability Engineer, Semantic Understanding

    San Jose, CA, USA

    • Staff Software Engineer, Display and Video 360 Inventory

    Kirkland, WA, USA; Mountain View, CA, USA

    • Technical Lead, Cluster Management System

    Warsaw, Poland

    • Staff Software Engineer, AI/ML GenAI, Google Cloud AI

    Sunnyvale, CA, USA; Kirkland, WA, USA

    • Customer Engineer, Platform, Start-ups, Google Cloud

    Singapore

    • Security Sales Specialist III, Google Cloud

    Sunnyvale, CA, USA; Atlanta, GA, USA; +6 more; +5 more

    • Senior Software Engineer, AI/ML, Ads and Commerce

    Mountain View, CA, USA

    • New Business Sales Account Strategist, Onboarding, Google Customer Solutions

    New York, NY, USA

    • Lab Operations Manager

    New Taipei, Banqiao District, New Taipei City, Taiwan

    • Senior Strategic Agency Manager, Google Customer Solutions (German Market)

    Dublin, Ireland

    • Program Manager III, Machine Delivery and Deployment, Cloud Supply Chain

    Atlanta, GA, USA

    • Product Manager, Core Experiences XR App

    San Jose, CA, USA

    • Business Program Manager, Platforms, Education Impact

    San Francisco, CA, USA; Mexico City, CDMX, Mexico; +4 more; +3 more

    Showing 1 to 20 of 3445 rows

    1‑20 of 3445

    navigate_next

    Go to next page

    Follow Life at Google on

    *

    More about us

    About usopen_in_newContact usopen_in_newPressopen_in_new

    Related Information

    Investor relationsopen_in_newBlogopen_in_new

    Equal Opportunity

    Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google"s EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

    More about us

    expand_more

    Related information

    expand_more

    Equal opportunity

    expand_more

    Privacyopen_in_newApplicant & Candidate Privacyopen_in_newTermsopen_in_newManage cookies

    helpHelpopen_in_new

    arrow_back

    Back to jobs search

    Staff Software Engineer, On-Device Machine Learning Infrastructure

    share

    Share Staff Software Engineer, On-Device Machine Learning Infrastructure

    • linkCopy link
    • emailEmail a friend

    corporate_fareGoogleplaceSunnyvale, CA, USA

    bar_chartAdvanced

    Advanced

    Experience owning outcomes and decision making, solving ambiguous problems and influencing stakeholders; deep expertise in domain.

    Apply

    share

    Share Staff Software Engineer, On-Device Machine Learning Infrastructure

    • linkCopy link
    • emailEmail a friend

    info_outline

    XIn 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 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).

    Preferred qualifications:

    • 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 working in a complex organization involving cross-functional, or cross-business projects.

    • Track record of leading and delivering ML projects focused on on-device deployment (e.g., Android, iOS, web browsers, or embedded devices).

    • Knowledge of ML converters/compilers and runtimes, and hardware-accelerated ML inference techniques.

    • Understanding of Generative AI model architectures and their optimization for on-device execution.

    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.

    With your technical expertise you will manage project priorities, deadlines, and deliverables. You will design, develop, test, deploy, maintain, and enhance software solutions.

    In this role you will deliver on-device ML infrastructure with leading performance, enabling framework and device optionality at scale. You will enable on-device deployment of key models, such as Gemini Nano and Gemma, across various accelerators (e.g., GPU / Pixel TPU / NPUs / CPU) on Android, Chrome, and more.The Google Cloud AI Research team addresses AI challenges motivated by Google Cloud's mission of bringing AI to tech, healthcare, finance, retail and many other industries. We work on a range of unique problems focused on research topics that maximize scientific and real-world impact, aiming to push the state-of-the-art in AI and share findings with the broader research community. We also collaborate with product teams to bring innovations to real-world impact that benefits our customers.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

    Learn more about benefits at Google.

    Responsibilities

    • Create roadmaps for developer-facing Application Programming Interfaces (APIs), Software Development Kits (SDKs), and tools, ensuring they meet the evolving needs of Large Language Models (LLMs) workflows.
    • Solve technically tests problems that exceed the scope of a generalist Software Engineers, specifically around optimizing Generative AI performance across heterogeneous hardware (CPUs, GPUs, and EdgeTPUs).
    • Guide the team in designing resilient and robust systems, proactively 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.

    Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit is subject to Google"s Applicant and Candidate Privacy Policy.

    Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google"s EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

    If you have a need that requires accommodation, please let us know by completing our Accommodations for Applicants form.

    Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

    To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

    Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc. Alphabet Inc. is your sole contractual partner with respect to equity grants. GSU grants are not guaranteed, are discretionary, are subject to approval by the Alphabet Inc. board of directors or its delegate, the terms of the relevant Alphabet Inc. stock plan, and your grant agreement. They have no impact on statutory payments. Current or past grants do not confer an acquired right.

    Follow Life at Google on

    *

    More about us

    About usopen_in_newContact usopen_in_newPressopen_in_new

    Related Information

    Investor relationsopen_in_newBlogopen_in_new

    Equal Opportunity

    Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google"s EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

    More about us

    expand_more

    Related information

    expand_more

    Equal opportunity

    expand_more

    Privacyopen_in_newApplicant & Candidate Privacyopen_in_newTermsopen_in_newManage cookies

    helpHelpopen_in_new

    Follow Life at Google on

    *

    More about us

    About usopen_in_newContact usopen_in_newPressopen_in_new

    Related Information

    Investor relationsopen_in_newBlogopen_in_new

    Equal Opportunity

    Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google"s EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

    More about us

    expand_more

    Related information

    expand_more

    Equal opportunity

    expand_more

    Privacyopen_in_newApplicant & Candidate Privacyopen_in_newTermsopen_in_newManage cookies

    helpHelpopen_in_new

    Google apps

    Main menu

    Numbers & Facts

    LocationSunnyvale, CA

    Skills

    • Algorithmsunmatched
    • Androidunmatched
    • Application Programming Interface (API)unmatched
    • Artificial Intelligence (AI)unmatched
    • Artificial Intelligence (AI) Natural Languageunmatched
    • Business Strategyunmatched
    • CPU (Central Processing Unit)unmatched
    • Cloud Computingunmatched
    • Cloud Storageunmatched
    • Computer Scienceunmatched
    • Cross-Functionalunmatched
    • Data Modelingunmatched
    • Data Processingunmatched
    • Data Storageunmatched
    • Data Structuresunmatched
    • Debugging Skillsunmatched
    • Distributed Computingunmatched
    • Embedded Systemsunmatched
    • English Languageunmatched
    • Equal Employment Opportunity (EEO)unmatched
    • Financeunmatched
    • GPU (Graphics Processing Unit)unmatched
    • Geneticsunmatched
    • Healthcareunmatched
    • Information Retrievalunmatched
    • Information/Data Security (InfoSec)unmatched
    • Internet Searchunmatched
    • Laboratory Operationsunmatched
    • Large-Scale Systemsunmatched
    • Leadershipunmatched
    • Machine Learningunmatched
    • Mentoringunmatched
    • Modeling Languagesunmatched
    • Natural Language Processing (NLP)unmatched
    • Network Designunmatched
    • Onboardingunmatched
    • Operations Managementunmatched
    • Performance Analysisunmatched
    • Performance Tuning/Optimizationunmatched
    • Polish Languageunmatched
    • Problem Solving Skillsunmatched
    • Product Managementunmatched
    • Product/Service Launchunmatched
    • Project/Program Managementunmatched
    • Recruiting/Staffing Agencyunmatched
    • Regulatory Requirementsunmatched
    • Reinforcement Learningunmatched
    • Reliability Engineeringunmatched
    • Retailunmatched
    • Salesunmatched
    • Sales Strategyunmatched
    • Software Administrationunmatched
    • Software Architectureunmatched
    • Software Architecture Designunmatched
    • Software Developmentunmatched
    • Software Engineeringunmatched
    • Startupunmatched
    • Stock Purchase Plansunmatched
    • Supply Chainunmatched
    • Technical Leadershipunmatched
    • Technical Supportunmatched
    • Test Plan/Scheduleunmatched
    • Testingunmatched
    • Time Managementunmatched
    • User Interface Designunmatched
    • Web Browsersunmatched
    • Workflow Analysisunmatched
    • YouTubeunmatched
    • iOSunmatched

    Be found by employers

    5,500+ employers search our resume database daily. Add yours to get found by recruiters looking for candidates like you.

    Level up your application

    Professional resume templates

    Browse dozens of recruiter approved resume templates, layouts and formats. Choose your favorite and make it your own in minutes.

    Free resume templates

    Free resume builder

    Improve your existing resume or start from scratch and create a standout, ATS-friendly resume. Add job-specific content, download and apply.

    Free resume builder