Staff Software Engineer, AI and Infrastructure

Google Google · Big Tech · Sunnyvale, CA +1

Staff Software Engineer role focused on building and scaling the core infrastructure (Borglet) for managing AI/ML workloads, including GPU/TPU management, within Google Cloud's enterprise AI platform. The role involves designing, implementing, and analyzing computer systems and their interactions with kernel and hardware to support hyperscale computing and advanced models.

What you'd actually do

  1. Design, implement, and analyze computer systems and their interactions with the kernel and hardware.
  2. Collaborate with partner teams as well as users across Google e.g., Borg team, ML teams, HW platform teams, SRE teams, Google's internal and Cloud users
  3. Solve ambiguous and high impact problems.
  4. Conduct strategic planning and tactical execution in complex projects. Ability to cross-coordinate across partner teams in Warsaw.
  5. Develop junior engineers on the team.

Skills

Required

  • C++
  • software products
  • large-scale infrastructure
  • distributed systems
  • networks
  • compute technologies
  • storage
  • hardware architecture
  • software design
  • software architecture

Nice to have

  • Linux Internals
  • Cluster Management
  • System Architecture
  • Virtualization
  • Security
  • technical leadership
  • project teams
  • technical direction
  • complex, matrixed organization
  • cross-functional projects
  • cross-business projects

What the JD emphasized

  • AI/ML Infrastructure
  • GPU/TPUs in cluster management system
  • hyperscale computing
  • Vertex AI
  • Gemini models

Other signals

  • AI/ML Infrastructure
  • GPU/TPUs in cluster management system
  • hyperscale computing
  • Vertex AI
  • Gemini models