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.