Experience with Large Language Models (LLMs) or Vision-Language Models (VLMs) in production.
Proven ability to optimize model performance on hardware accelerators (TPUs, GPUs).
Familiarity with machine learning frameworks, especially JAX or TensorFlow, for inference and serving.
Responsibilities
Architect scalable full-stack model serving solutions for robotics, integrating on-robot and cloud systems.
Conduct performance analysis and optimization across the sensory-to-action loop in robotics.
Collaborate with research scientists and engineers to design flexible and reliable deployment workflows.
Benefits
15% bonus target equity benefits.
Comprehensive health and wellness programs.
Opportunities for professional development and training.
Flexible work arrangements and supportive work culture.
Full Job Description
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 .
Responsibilities
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.
Individual pay is determined by factors including job-related skills, experience, and relevant education or training.
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.
