You will thrive at the intersection of machine learning infrastructure, distributed systems, and robotics, managing complex end-to-end challenges from co-designing on-robot execution and bidirectional streaming pipelines to optimizing and scaling foundation model inference across edge and cloud accelerators.
Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high-quality product innovation for billions of users. We use our technologies for widespread public benefit and scientific discovery, ensuring safety and ethics are always our highest priority.
- We are pushing the boundaries across multiple domains. Our global teams offer diverse learning opportunities and varied career pathways for those driven to achieve exceptional results through collective effort.
- Individual pay is determined by factors including job-related skills, experience, and relevant education or training.
US: $174000 - $252000 (USD) + 15% bonus target + equity + benefits
Learn more about benefits at Google.
Architect highly reliable and scalable full-stack model serving and deployment solutions spanning on-robot runtimes, local edge compute, and cloud backend systems (such as Beyond and Uniserve) customizing for stateful bidirectional streaming, multi-rate agentic control loops, and fast model hot-swapping in real-world robotics deployments.
Conduct end-to-end performance analysis, profiling, and optimization across the entire sensory-to-action loop, from on-robot sensor ingestion, transport, and real-time execution handoffs to KV-cache management, graph compilation, and accelerator inference for cutting-edge robotics foundation models.
Partner closely with research scientists, hardware/control engineers, and organizational partner teams to co-design agentic harnesses, streaming APIs, and closed-loop deployment workflows that provide flexibility and reliability across various robotics applications.
Minimum qualifications
Bachelor's degree or an advanced degree.
4 years of experience in software development.
Experience working with Large Language Models (LLMs) or Vision -Language Models (VLMs) in a production environment.
Experience optimizing model performance for inference on hardware accelerators (e.g., TPUs and GPUs).
Experience with machine learning frameworks, particularly JAX or TensorFlow, focused on inference and serving.
Preferred qualifications
Experience developing and integrating Google’s production model serving solutions like Beyond and Uniserve.
Experience integrating security best practices directly into ML infrastructure.
Expertise in building, profiling, and optimizing complex, highly available, and secure software systems for end-to-end performance.
Familiarity with robotics software stacks and demonstrated experience taking complex technical features from initial design through to production deployment.
Strong, proven programming skills in Python and C++.
Bachelor's degree or an advanced degree.
4 years of experience in software development.
Experience working with Large Language Models (LLMs) or Vision -Language Models (VLMs) in a production environment.
Experience optimizing model performance for inference on hardware accelerators (e.g., TPUs and GPUs).
Experience with machine learning frameworks, particularly JAX or TensorFlow, focused on inference and serving.
