Jobs · Engineering · California

Senior Engineering Manager AI Inference Platform, Distributed Cloud

Google · Sunnyvale, CA · 6 days ago
On-siteEngineeringFull-time

About the job

In this role, you will be pivotal in architecting and optimizing the serving stack for models like Gemini in an on-prem cloud environment, addressing exciting challenges to improve speed and efficiency. This is a unique opportunity to go deep, leading system-level design and performance profiling, ensuring Google's LLMs run faster and more cost-effectively than ever before.

Responsibilities

  • Lead, mentor, and grow a high-performing team of systems and ML engineers. Drive a culture of excellence, psychological safety, and continuous learning while guiding career paths and OKRs.
  • Define the technical vision and strategy for enhancing the LLM serving stack, focusing on performance, scalability, and resource efficiency.
  • Oversee the infrastructure and tooling for in-depth performance analysis, profiling, and benchmarking of LLM models on GPU accelerators.
  • Partner closely with Research, SRE, Product, and core library teams to optimize and deploy LLMs globally.
  • Drive the design, implementation, and optimization of advanced serving architectures—including disaggregated serving—while collaborating with core library and kernel partners to eliminate low-level performance bottlenecks, maximize resource utilization, and minimize latency.

Qualifications

  • Minimum qualifications:
  • - Bachelor's degree or equivalent practical experience.
  • - 8 years of experience programming in C++ or Python.
  • - 7 years of experience optimizing, profiling, and scaling production-grade systems on GPU accelerators or specialized AI hardware.
  • - 5 years of experience directly managing and leading engineering teams focused on machine learning infrastructure, AI platforms, or high-performance distributed computing systems.
  • - 5 years of experience in a people management or team leadership role.
  • - 4 years of experience managing engineering organizations across multi-team infrastructure dependencies.
  • Preferred qualifications:
  • - Master’s degree or PhD in Engineering, Computer Science, or a related technical field.
  • - 5 years of experience working in a complex, matrixed organization.
  • - 5 years of experience implementing advanced LLM serving architectures and optimization techniques, such as disaggregated serving, continuous batching, or specialized compiler technologies (e.g., XLA).
  • - 4 years of experience utilizing deep-dive ML profiling tools (e.g., Nsight, xprof) to troubleshoot and resolve low-level bottlenecks within major frameworks like JAX, PyTorch, or TensorFlow.

Pay

US: $262000 - $365000 (USD) + 25% bonus target + equity + benefits

Schedule

Full-time

Benefits

Learn more about benefits at Google.

Similar jobs

Senior Cloud Engineer AI

BMO U.S.Jacksonville, FL· 6 days ago
Engineering$81k–$152k/yrapply on bmo.wd3.myworkdayjobs.com