Staff Software Engineer, Tpu Machine Learning Supercomputer

Google Google · Big Tech · Sunnyvale, CA +1

Staff Software Engineer on the TPU Machine Learning Supercomputer (MLSC) team, responsible for designing and developing features to improve the scalability and reliability of large-scale software across TPUs and other distributed networked hardware machines. The role involves working on various layers of the software stack, from host daemons to network routing and distributed control software, and providing technical leadership for future supercomputer generations. The team focuses on delivering AI and Infrastructure at scale, efficiency, reliability, and velocity, supporting Google's groundbreaking innovations and hyperscale computing.

What you'd actually do

  1. Design, develop, test, deploy, and debug critical system software that enables TPU Machine Learning accelerators to function seamlessly.
  2. Develop advanced analytics and health management capabilities to effectively manage and optimize large-scale ML systems.
  3. Lead high-impact projects and steer successful delivery while ensuring alignment with broader team strategies.
  4. Provide technical guidance and mentorship to software engineers to foster their professional growth and development.

Skills

Required

  • software development in C++ or Go
  • large-scale infrastructure
  • distributed systems
  • networks
  • testing and launching software products
  • software design and architecture
  • operating systems
  • data structures
  • algorithms
  • developing, integrating, and testing system and user-space software for hardware accelerators or TPU systems

Nice to have

  • data structures and algorithms
  • technical leadership role leading project teams and setting technical direction
  • complex, matrixed organization involving cross-functional, or cross-business projects
  • building backend software for high-performance computing (HPC) and machine learning (ML) applications
  • data analytics
  • ML architecture
  • how common algorithms map to software/hardware operations
  • highly distributed systems
  • control plane and management Software
  • networking concepts

What the JD emphasized

  • critical system software
  • large-scale ML systems
  • high-impact projects
  • technical leadership

Other signals

  • TPU Machine Learning Supercomputer
  • large-scale software across TPUs
  • distributed networked hardware machines
  • host daemons to network routing and distributed control software
  • future supercomputer generations
  • AI and Infrastructure team
  • unparalleled scale, efficiency, reliability and velocity
  • essential platforms that enable developers to build the future
  • world-leading hyperscale computing
  • TPUs, Vertex AI for Google Cloud, Google Global Networking, Data Center operations, systems research