Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff Software Engineer, TPU Machine Learning Supercomputer

$207k - $300k

Google

Minimum Qualifications Bachelor's degree or equivalent practical experience. 8 years of experience with software development in C++ or Go. 5 years of experience with large‑scale infrastructure, distributed systems, or networks, as well as testing and launching software products. 3 years of experience with software design and architecture. Experience with operating systems, data structures, and algorithms. Experience developing, integrating, and testing system and user‑space software (including tools, dashboards, and monitoring) for hardware accelerators or TPU systems. 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 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 building backend software for high‑performance computing (HPC) and machine learning (ML) applications, including knowledge of data analytics, ML architecture, and how common algorithms map to software/hardware operations. Understanding of highly distributed systems, control plane and management software, and networking concepts. 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. As a member of the TPU Machine Learning Supercomputer (MLSC) team, you will design and develop features to significantly improve the scalability and reliability of large‑scale software across TPUs and other distributed networked hardware machines. Your work will span various layers of the software stack, from host daemons to network routing and distributed control software running across Google's internal and cloud infrastructure. You will also provide leadership to help formulate and drive software development plans for future supercomputer generations. The AI and Infrastructure team is redefining what’s possible. We empower Google customers with breakthrough capabilities and insights by delivering AI and Infrastructure at unparalleled scale, efficiency, reliability, and velocity. Our customers include Googlers, Google Cloud customers, and billions of Google users worldwide. We're the driving force behind Google's groundbreaking innovations, empowering the development of our cutting‑edge AI models, delivering unparalleled computing power to global services, and providing the essential platforms that enable developers to build the future. From software to hardware, our teams are shaping the future of world‑leading hyperscale computing, with key teams working on the development of our TPUs, Vertex AI for Google Cloud, Google Global Networking, Data Center operations, systems research, and much more. The US base salary range for this full‑time position is $207,000‑$300,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. Within the range, individual pay is determined by work location and additional factors, including job‑related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your preferred location during the hiring process. Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits. Learn more about benefits at Google. Responsibilities Design, develop, test, deploy, and debug critical system software that enables TPU Machine Learning accelerators to function seamlessly. Develop advanced analytics and health management capabilities to effectively manage and optimize large‑scale ML systems. Lead high‑impact projects and steer successful delivery while ensuring alignment with broader team strategies. Provide technical guidance and mentorship to software engineers to foster their professional growth and development. Equal Employment Opportunity Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form. #J-18808-Ljbffr Google

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Staff Software Engineer, TPU Machine Learning Supercomputer in Sunnyvale, CA vacancy
  • $207k - $301k

     ...equivalent practical experience. 8 years of software development experience in C, C++, Go, or...  ...Master’s degree or PhD in Engineering, Computer Science, or a related technical...  ...target + equity + benefits Responsibilities Learn more about benefits at Google. Lead the... 
    Suggested
    Remote work
    Worldwide

    Google

    Sunnyvale, CA
    2 days ago
  • Staff Software Engineer, Machine Learning - Personalization San Francisco, CA; Sunnyvale, CA About the Team Come help us build the world's most reliable on-demand, logistics engine for last-mile retail delivery! We're looking for an experienced machine learning engineer... 
    Suggested
    Work at office
    Local area
    Remote work

    DoorDash USA

    Sunnyvale, CA
    2 days ago
  • $262k - $365k

    Senior Staff Software Engineer, GeminiApp Gen Media, DeepMind Mountain View, CA, USA Minimum qualifications...  ...the world’s first general‑purpose learning agent. Central to this mission is the...  ...developed by our exceptional team of Machine Learning and Neuroscience research... 
    Suggested

    Google

    Mountain View, CA
    2 days ago
  • $189k - $284k

     ...experience. 8 years of experience in software development, and with data structures/...  ...architecture. 5 years of experience in the Machine Learning field. 5 years of coding experience...  ...: Master’s degree or PhD in Engineering, Computer Science, or a related technical... 
    Suggested
    Full time

    Google

    Mountain View, CA
    2 days ago
  • $207k - $301k

     ...or equivalent practical experience. 8 years of experience in software development. 5 years of experience testing and launching software...  ...). Preferred qualifications Master’s degree or PhD in Engineering, Computer Science, or a related technical field. 8 years of experience... 
    Suggested

    Google

    Mountain View, CA
    2 days ago
  • About the role: We are seeking senior or staff software engineers to join our planning team to build motion planning and decision‑making systems...  ...in applied research teams Experience in applying machine learning for planning problems (e.g. Imitation Learning, Behavior... 
    Work at office
    Remote work

    Gatik AI, Inc.

    Mountain View, CA
    2 days ago
  •  ...streamline complex workflows, and continuously learn and adapt. Moveworks is trusted by over...  ...automation with Moveworks’ Reasoning Engine and natural language capabilities, we...  ...solve real world problems by leveraging our machine learning platform, build the foundational... 
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Mountain View, CA
    2 days ago
  • About the Role We’re looking for a passionate Applied Machine Learning expert to join our team. As a Staff Machine Learning Engineer, you’ll be conceptualizing, designing, implementing, and validating algorithmic improvements to the catalog system and our product knowledge... 
    Full time
    Temporary work
    Work at office
    Local area
    Remote work
    Flexible hours

    DoorDash

    Sunnyvale, CA
    2 days ago
  • $189k - $284k

     ...experience. 8 years of experience in software development, and with data structures/...  ...development. 2 years of experience in Machine Learning (ML), ML Efficiency, or a related field...  ...qualifications Master’s degree or PhD in Engineering, Computer Science, or a related... 
    Full time

    Google

    Mountain View, CA
    2 days ago
  • $240k - $265k

    Staff Software Engineer, Learned & Hybrid Behavior Planning Mountain View, CA Kodiak Robotics, Inc. was founded in 2018 and has become a leader in...  ...this role, you will sit at the intersection of Planning and Machine Learning, working closely with ML engineers and autonomy... 
    Visa sponsorship

    Kodiak

    Mountain View, CA
    2 days ago
  •  ...Systems in Sunnyvale, CA is seeking a Member of Technical Staff (Software Engineer) to implement infrastructure for high-performance, low-...  ..., optimizing resource allocation, and collaborating with machine learning engineers. Strong skills in Docker, Python, and infrastructure... 

    Cerebras Systems

    Sunnyvale, CA
    1 day ago
  • $181k - $297k

     ...Governance, Risk, Automation, Compliance and Engineering. Our commitment to our customers and...  ...deep expertise in data modeling, machine learning, and advanced analytics to ensure...  ...data foundations in large, distributed software environments. Experience creating data... 
    For contractors
    Work experience placement
    Work at office
    Flexible hours
    Shift work

    LinkedIn

    Mountain View, CA
    2 days ago
  • Member of Technical Staff (Software Engineer) Sunnyvale, CA Cerebras Systems builds the world’s largest AI chip, 56 times larger than GPUs...  ...industry‑leading training and inference speeds, empowering machine learning users to run large‑scale ML applications without managing... 
    Full time
    Part time
    Internship

    Cerebras Systems

    Sunnyvale, CA
    1 day ago
  • $143k - $286k

    Position Summary... As a Staff Software Engineer, you’ll be a technical leader who defines the direction...  ...using classical ML, NLP, and deep learning techniques. Autocomplete - Real‑time...  ...relevance optimization. Background in machine learning - including core concepts (classification... 
    Full time
    Temporary work
    Part time
    Work at office
    Flexible hours

    Walmart

    Sunnyvale, CA
    2 days ago
  • Job Description The Role Are you a software engineer who has honed your craft, partnership, and execution over years building and improving...  ...productionize them at scale Use the latest advances in machine learning, LLMs, and AI agents to enhance our products and create... 
    Immediate start

    ServiceNow

    Mountain View, CA
    2 days ago
  • $207k - $300k

     ...experts in AI, energy systems, software engineering and product design—all...  ...we would love to meet you. Learn more about our team and our...  ...mission here. About the role As a staff backend software engineer...  ...with other software engineers, machine learning experts, power... 
    Full time
    Flexible hours

    Tapestry

    Mountain View, CA
    2 days ago
  • $207k - $301k

     ...years of experience with software development in one or more...  ...Experience with computer vision learning techniques, machine learning algorithms (...  ...Experience working in large-sized engineering and cross‑functional teams...  ...to that tomorrow. As a Staff Software Engineer, you... 

    Google

    Mountain View, CA
    2 days ago
  • $143k - $286k

    Position Summary We are looking for an experienced Staff Software Engineer to join Walmart Search Runtime Engineering team. This hands‑on...  ...and more. By combining cutting‑edge information retrieval, machine learning, and ranking algorithms, we build low‑latency, high‑... 

    Walmart

    Sunnyvale, CA
    2 days ago
  • $235.03k - $352.29k

     ...richer, and more connected future. About the Role Nuro takes a machine‑learning‑first approach to autonomous driving technology. In an ML‑...  ...development. The team closely collaborates with autonomy engineers to ensure our labeled data is high‑quality and comprehensive... 

    Nuro

    Mountain View, CA
    2 days ago
  • $193.93k - $352.29k

     ...connected future. About the Role Nuro takes a machine‑learning‑first approach to autonomous driving...  ...team collaborates closely with system engineers to thoroughly validate the autonomous...  ...and best practices across broader software organization A bachelor's degree in Computer... 
    Work experience placement

    Nuro

    Mountain View, CA
    2 days ago
  • $192.6k - $288.8k

     ...performance, reliability, and growth across our ecosystem. As a Staff Software Engineer on the Feature Platform team, you'll play a critical role in building the infrastructure that powers machine learning, experimentation, and optimization across our ads ecosystem. You... 
    Work at office
    Worldwide
    Relocation package

    Unity South APAC (SEA, ANZ, IND Subcont.)

    Mountain View, CA
    2 days ago
  • $248k - $349k

    Senior Staff Software Engineer, Agent Protocol, Cloud AI 2 days ago Be among the first 25 applicants Get AI-powered advice on this job and...  ...in Computer Science, Mathematics, Applied Statistics, Machine Learning, or equivalent practical experience. 8 years of experience... 
    Full time

    Google

    Sunnyvale, CA
    2 days ago
  • $131k - $285k

     ...most reliable on-demand logistics engine. Behind the scenes, our Machine Learning Platform (MLP) powers critical...  ...the Role We’re looking for a Staff Software Engineer with deep expertise in ML...  ...across heterogeneous hardware (GPU/TPU/CPU), with strong isolation,... 
    Hourly pay
    Work at office
    Local area
    Remote work
    Flexible hours

    DoorDash

    Sunnyvale, CA
    more than 2 months ago
  • $160k - $230k

     ...maintenance of Plus' autonomous onboard software infrastructure. You will work...  ...areas such as computer vision, robotics, machine learning, embedded systems, mechatronics, and functional...  ...the high performance C++ runtime engine that powers the entire autonomous driving... 

    PlusAI, Inc.

    Santa Clara, CA
    2 days ago
  • $251k - $310k

     ...with. We do this by improving onboard software architecture, system debuggability, and...  ...of leading complex, cross-functional engineering projects as a technical lead. Deep understanding...  ...Master's or PhD in Computer Science, Machine Learning, Robotics, or a related field.... 
    Full time
    Remote work
    Flexible hours

    Waymo

    Mountain View, CA
    2 days ago
  • $197k - $291k

    Staff Software Engineer, AI/ML GenAI, Google Cloud Location: Sunnyvale, CA, USA; Washington D.C., DC, USA Advanced Experience owning outcomes and...  ...salary only, and do not include bonus, equity, or benefits. Learn more about benefits at Google. Responsibilities Design,... 
    Full time

    Google

    Sunnyvale, CA
    2 days ago
  •  ...detect cancer. Employing advanced machine-learning methods to whole-genome...  ...major systems with a focus on software and cloud architecture. You'll partner across Engineering, Data Science, R&D, Regulatory...  ...software components. Partner with staff engineers on test design and... 
    Flexible hours
    3 days per week

    DELFI Diagnostics

    Palo Alto, CA
    2 days ago
  • $197k - $291k

     ...years of experience testing, and launching software products, and 3 years of experience with...  .... About The Job Google's software engineers develop the next-generation technologies...  ...not include bonus, equity, or benefits. Learn more about benefits at Google . Responsibilities... 
    Full time

    Google

    Mountain View, CA
    5 days ago
  • $262k - $365k

    Senior Staff Software Engineer, Google Compute Engine (Guest Ecosystem) Locations: Sunnyvale, CA, USA...  ...SAP and AI/ML workloads via Virtual Machine (VM) Extensions. The Guest Ecosystem...  ...5% bonus target + equity + benefits. Learn more about benefits at Google. Responsibilities... 
    Temporary work

    Google

    Sunnyvale, CA
    2 days ago
  • $207k - $300k

     ...practical experience. 8 years of experience in software development. 5 years of experience...  ...: Master’s degree or PhD in Engineering, Computer Science, or a related technical...  ...not include bonus, equity, or benefits. Learn more about benefits at Google . Responsibilities... 
    Full time
    Temporary work

    Google

    Sunnyvale, CA
    2 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Software Engineer, TPU Machine Learning Supercomputer. Be the first to apply!