Post your job offer for free on H1BConnect with no upfront cost!

Logo

Hire with Us
Google logo

Staff Software Engineer, GKE AI Agent Platform

Google

10/15/2025

San Francisco, CA

Full-time

Salary: $197,000 - $291,000 a year


Job Description

Google is seeking a talented software engineer to work on next-generation technologies in the field of artificial intelligence and cloud computing.

Requirements

  • Bachelor's degree or equivalent practical experience.
  • 8 years of experience in software development.
  • 5 years of experience in cloud computing and building operating systems.
  • Experience delivering in hypergrowth and changing development environments, ideation, and innovating technology.
  • Experience with operating a cloud product.
  • Experience in statistical analysis and classical machine learning.
  • Experience in contributing to open source software.
  • Understanding of building solutions or contributing to Kubernetes.
  • Ability to possess a developed technical perspective and architecture and system design.
  • Track record of building Machine Learning (ML) or Generative Artificial Intelligence (Gen AI) agentic platform.

Responsibilities

  • Lead the design of a unified Google Kubernetes Engine (GKE) architecture supporting both Reinforcement Learning Fine-Tuning (RLFT) and Agent-Level RL workloads.
  • Drive end-to-end performance improvements of the RL step time. Address bottlenecks in sampler inference servers, storage I/O with a focus on low-latency Pod Snapshots, and high-throughput networking for terabyte-scale weight synchronization.
  • Oversee cross-team initiatives to build a vertically integrated solution.
  • Investigate and prototype foundational GKE capabilities for agent forensics and observability. Evaluate purpose-built AI hardware to create a defensible performance advantage for RL on GKE.
  • Establish GKE as the premier platform for the Open Source Software (OSS) AI ecosystem. Publish performance benchmarks and reference architectures for leading Reinforcement Learning (RL) libraries.

Benefits

  • Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.
Logo

© 2024 H1BConnect. All rights reserved.

Check out our sister site LatamDev for tech jobs in Latin America! 🌎