Software Engineer III, TPU Supercomputer, Cloud Platforms

Google•Apply•Published 20 hours ago•First seen 20 hours ago

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.

In this role, you will develop software that enables TPU, Google’s custom-built AI computation chip, to run large-scale AI hypercomputation in Google’s data centers, and thus empowering all the cutting edge AI innovations for Google (Deepmind, Search, Ads) and other Cloud customers.

You will cover a broad range of software in the TPU software stack, including system software that enable a single TPU machine, superpod software that connects thousands of TPU chips into a AI hypercomputer, and health monitoring software that ensures the TPUs and their interconnection and networking are healthy, etc.

You will play a key role in the introduction of each new TPU chip, from design, system bringup, to productionization of individual machines.

Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google’s cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.

Responsibilities

Design and implement software across the entire TPU supercomputer stack.
Own and lead significant design and implementation efforts for Google's Machine Learning Supercomputer infrastructure.
Develop API extensions and integrations for user-facing applications and orchestrations systems.
Create comprehensive functional and integration tests for all developed software components.
Design and build advanced diagnostics and monitoring tools.

Minimum qualifications:

Bachelor's degree in Electrical Engineering, Computer Science, a related technical field, or equivalent practical experience.
2 years of experience with software development in C++.

Preferred qualifications:

Master's degree or PhD in Electrical Engineering, Computer Science, or in related technical fields.
Experience with data structures or algorithms in either an academic or industry setting.
Experience in developing software for network control planes or high-performance interconnects.
Experience in developing or integrating with cluster management systems like cloud platforms such as Google Kubernetes Engine (GKE).
Experience with building, monitoring, diagnostics, and testing frameworks for complex, distributed systems.
Knowledge of network topologies and protocols.