Google

Staff AI/ML Engineer, AI Rapid Response Team

Google$207K — $300K *
Information Technology
8 - 10 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's degree or equivalent practical experience.
  • 8 years of software development experience.
  • 5 years in testing and launching software products; 3 years in software design and architecture.
  • 5 years of experience with speech/audio, reinforcement learning, or ML infrastructure.
  • 5 years with ML design and infrastructure such as model deployment and debugging.
  • Experience integrating generative AI tools or LLM interfaces.

Responsibilities

  • Lead technical architecture and system design for short-term engineer embeds and sprints.
  • Design, prototype, and produce robust production code in C and Python for model efficiency.
  • Diagnose and optimize latency and throughput issues in complex ML clusters.
  • Translate executive directives into actionable efficiency scopes with performance benchmarks.
  • Develop automated pipelines to optimize large foundation models for practical applications.

Benefits

  • Comprehensive health, wellness, and retirement plans.
  • Generous paid time off and flexible work arrangements.
  • Access to professional development initiatives and courses.
  • Relocation assistance for eligible candidates.
  • Opportunities for equity and performance bonuses.
Full Job Description
Minimum qualifications:
  • Bachelor's degree or equivalent practical experience.
  • 8 years of experience in software development.
  • 5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture.
  • 5 years of experience with one or more of the following: speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field.
  • 5 years of experience with ML design and ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning).
  • Experience integrating generative AI tools or LLM interfaces into workflows.

Preferred qualifications:
  • Master's degree or PhD in Engineering, Computer Science, or a related technical field.
  • 8 years of experience with data structures and algorithms.
  • 3 years of experience in a technical leadership role leading project teams and setting technical direction.
  • Experience with low-level accelerator programming and compilation toolchains (e.g., XLA, Pallas, CUDA, Triton, or custom TPU kernels).
  • Proven ability to lead rapid prototyping pods (SWAT/FDE) in highly ambiguous environments and influence VP/Director-level technical roadmaps.
  • Demonstrated track record of optimizing large-scale production ML serving or training systems resulting in measurable, compute/cost savings.


About the job

With your technical expertise you will manage project priorities, deadlines, and deliverables. You will design, develop, test, deploy, maintain, and enhance software solutions.

On the AI Rapid Response Team, you will serve as the primary technical anchor and chief efficiency strategist for Google's most demanding AI workloads. You will operate as an "ambiguity buster," taking nebulous VP-level mandates around compute constraints, latency spikes, and infrastructure scaling costs, and translating them into crisp, mathematically validated engineering solutions.

Trading long-term maintenance of legacy systems for continuous zero-to-one pathfinding velocity, you will lead Strike Sprints and embedded FDE engagements across Google. You will write high-performance production code, architect novel model efficiency pipelines, optimize inference serving engines, and establish graceful exit architectures that empower partner teams to run permanently lean.

Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $207000 - $300000 (USD) 20% bonus target equity benefits

Learn more about benefits at Google .

Responsibilities
  • Lead technical architecture and system design for complex 1-6-month Forward Deployed Engineer (FDE) embeds and 2-4-week Strike Sprints targeting high-leverage Machine Learning efficiency bottlenecks.
  • Design, prototype, and write robust production C and Python code for model compression, speculative decoding engines, dynamic batching layers, and high-throughput serving pipelines.
  • Diagnose subtle distributed latency and throughput bottlenecks across XManager, Pathways, and Tensor Processing Unit/Graphics Processing Unit clusters, implementing low-level kernel and memory optimizations (accelerated linear algebra (XLA), Pallas, Custom Ops).
  • Deconstruct ill-defined executive mandates ("The Hot Plate") into rigorous efficiency scopes within strict latency, floating point operations (FLOPs), and tokenomics thresholds-delivering compelling thinnest viable proofs (TVPs).
  • Design automated distillation, pruning, and quantization (FP8/INT4) pipelines that convert massive foundation models into compact, ultra-efficient student models without compromising evaluation benchmarks.


About Google

Google is a multinational technology company that specializes in Internet-related services and products. These include online advertising technologies, search engine, cloud computing, software, and hardware. Google was founded in 1998 by Larry Page and Sergey Brin while they were Ph.D. students at Stanford University. The company has grown tremendously since then and has become one of the most valuable companies in the world. Google's mission is to organize the world's information and make it universally accessible and useful.
Learn more about Google
Size
156,500 employees
Market Cap
$1,115.4 billion
Industry
Net Income
$40.2 billion
Founded
1998
5 Year Trend
+23.3%
Revenue
$182.5 billion
NASDAQ

Similar Jobs

More Jobs at Google

More Information Technology Jobs

Find similar Staff AI/ML Engineer, AI Rapid Response Team jobs: