16 hours ago
Mountain View, CA, USAMid Level
H1B sponsor
Base Salary
$174k - $252k/yr
Responsibilities
- Architect reliable and scalable model-serving and deployment solutions spanning on-robot runtimes, edge compute, and cloud systems.
- Customize serving systems for stateful bidirectional streaming, multi-rate agentic control loops, and rapid model hot-swapping in robotics deployments.
- Analyze, profile, and optimize the full sensory-to-action pipeline, including sensor ingestion, transport, real-time execution, KV-cache management, graph compilation, and accelerator inference.
- Partner with research scientists, hardware and control engineers, and organizational teams to co-design agentic harnesses, streaming APIs, and closed-loop deployment workflows.
- Design, develop, test, deploy, maintain, and enhance software while managing project priorities, deadlines, and deliverables.
Requirements
- Bachelor's degree or an advanced degree.
- Four years of experience in software development.
- Experience working with Large Language Models or Vision-Language Models in a production environment.
- Experience optimizing model inference performance on hardware accelerators such as TPUs and GPUs.
- Experience with machine learning frameworks, particularly JAX or TensorFlow, for inference and serving.
- Preferred: experience with Google's Beyond and Uniserve production model-serving solutions.
- Preferred: experience integrating security best practices into ML infrastructure.
- Preferred: expertise building, profiling, and optimizing highly available and secure software systems for end-to-end performance.
- Preferred: familiarity with robotics software stacks and taking complex technical features from design through production deployment.
- Preferred: strong programming skills in Python and C++.
Benefits
- US base salary of $174000-$252000, plus a 15% bonus target, equity, and benefits.
Categories
ML EngineeringRobotics
About Google
Google builds consumer and enterprise software and services including Search, Android, YouTube, Chrome, Maps, Gmail, and Google Cloud. Its business model centers on digital advertising and paid cloud, software, and hardware offerings (e.g., Pixel and Nest) for consumers, developers, and organizations. Founded in 1998 and headquartered in Mountain View, California, Google operates globally as a subsidiary of Alphabet Inc.