Machine Learning Engineer - Model Optimization

Who We Are Nuro is at the forefront of self-driving technology, dedicated to making autonomous driving accessible to everyone. Established in 2016, we are engineering the world’s most scalable driving solution by merging advanced AI with automotive-grade hardware. Our proprietary technology, the Nuro Driver™, is licensed to support a variety of applications, from robotaxis and commercial fleets to personal vehicles. With years of successful self-driving deployments, Nuro provides automakers and mobility platforms a clear pathway to achieving commercial-scale autonomous vehicles, fostering a safer, more connected future. About the Work Design and enhance state-of-the-art generative models, particularly focusing on diffusion architectures, flow-matching techniques, and energy-based models for autonomous planning. Develop generative models utilizing foundation models. Harness large language models and world foundation models for reasoning, decision-making, and multi-modal generation. Optimize generative models through reinforcement learning to enhance interactive reasoning. Investigate reward modeling and learned verifiers using generative models. Explore joint prediction and planning as well as self-play, and leverage generative models for active learning and world modeling. Create controllable generative models to direct the generation process towards specific goals, conditions, and rewards. Collaborate with autonomy teams to propose and implement holistic solutions to pressing autonomy challenges. Assess issues, suggest solutions, prioritize tasks, and evaluate your findings by deploying models on the Nuro Driver.

About UsNuro is pioneering self-driving technology with a mission to make autonomy accessible for everyone. Established in 2016, we are developing the world's most scalable driver, integrating advanced AI with automotive-grade hardware. Our flagship technology, the Nuro Driver™, is licensed to facilitate diverse applications, ranging from robotaxis to commercial fleets and personal vehicles. With years of proven deployment in self-driving environments, Nuro offers automakers and mobility platforms a viable pathway to commercial-scale autonomous vehicles, creating a safer, more connected future.Role OverviewAs a Senior/Staff Machine Learning Research Scientist, you will work closely with multidisciplinary teams, focusing on generative modeling to solve complex planning challenges in autonomous driving. You will utilize cutting-edge techniques—including diffusion models, flow matching, and energy-based models—to create innovative solutions that enable safe and efficient driving behavior in real-world scenarios. Additionally, you will manage the complete lifecycle of your models, transitioning them into robust applications for global autonomous driving deployments.Key ResponsibilitiesDesign and enhance state-of-the-art generative models, particularly focusing on diffusion architectures, flow-matching methods, and energy-based models, aimed at autonomous plan generation.Integrate large language models and world foundation models to facilitate reasoning, decision-making, and multi-modal generation.Employ reinforcement learning to optimize generative models for interactive reasoning, and investigate reward modeling and self-play methodologies.Create controllable generative models that steer the generation process towards specific goals and conditions.Collaborate with various autonomy teams to develop comprehensive solutions for key challenges in autonomous technology, ensuring rigorous evaluation through deployment on the Nuro Driver.

Waymo is at the forefront of autonomous driving technology, committed to becoming the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, we have dedicated ourselves to developing the Waymo Driver—The World’s Most Experienced Driver™. Our mission is to enhance mobility access while significantly reducing traffic-related fatalities. With over ten million rider-only trips powered by the Waymo Driver, we have accumulated extensive experience, having driven more than 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.The Waymo AI Foundations team is dedicated to creating machine learning solutions that tackle complex challenges in autonomous driving. Our goal is to ensure the safe operation of Waymo vehicles in various cities and under diverse driving conditions. We actively collaborate with other research teams within Alphabet, focusing on areas such as reinforcement learning, learning from demonstration, generative modeling, Bayesian inference, hierarchical learning, and robust evaluation.This position follows a hybrid work schedule and reports directly to a Staff Research Scientist / Tech Lead Manager.

Waymo, the leader in autonomous driving technology, aims to be the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, Waymo is dedicated to developing the Waymo Driver—The World's Most Experienced Driver™—to enhance mobility access and significantly reduce traffic-related fatalities. With over ten million rider-only trips facilitated by our technology, which has autonomously navigated more than 100 million miles on public roads and countless simulations across more than 15 U.S. states, we are at the forefront of autonomous transportation.The Waymo Applied Research team is devoted to crafting machine learning solutions that tackle key challenges in autonomous driving, focusing on safely operating Waymo vehicles across diverse cities and under varying driving conditions. We thrive on collaboration, fostering partnerships with other research teams within Alphabet to push the boundaries of innovation.In this hybrid role, you will report to a Technical Lead Manager.

About UsNuro is a pioneering self-driving technology firm committed to making autonomy accessible to everyone. Established in 2016, we are developing the world's most scalable autonomous driver by integrating advanced artificial intelligence with automotive-grade hardware. Nuro licenses its proprietary technology, the Nuro Driver™, for diverse applications including robotaxis, commercial fleets, and personal vehicles. With years of proven technology through self-driving deployments, we provide automakers and mobility platforms with a clear pathway to commercial-scale autonomous vehicles, fostering a safer, more interconnected future.Role OverviewAs a Staff ML Research Scientist, you will collaborate with our autonomy teams to tackle perception, prediction, and planning challenges using innovative machine learning techniques. Your expertise in foundation models, multi-modal LLMs, generative models, out-of-distribution detection, imitation learning, and reinforcement learning will be invaluable. If you are passionate about solving complex problems and eager to see your solutions implemented in real-world scenarios, we invite you to join our team!Key ResponsibilitiesConduct research and apply cutting-edge machine learning methodsAddress state-of-the-art challenges in perception, prediction, and planningDevelop and optimize efficient ML data pipelinesLead and mentor a diverse team of professionalsProduce and deploy autonomous software for public deployment

Waymo is at the forefront of autonomous driving technology, dedicated to becoming the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, our focus has been on developing the Waymo Driver—The World’s Most Experienced Driver™—to enhance mobility access and save lives currently lost in traffic incidents. With over ten million rider-only trips facilitated by our technology, and more than 100 million miles driven autonomously on public roads, we are transforming transportation across 15+ U.S. states.The Predictive Planning team (PrePlan) is responsible for designing and implementing advanced machine learning solutions that forecast future world states and plan the Waymo Driver’s actions. Our goal is to harness Waymo’s extensive driving data to create robust, adaptable, and efficient deep neural networks that empower our autonomous vehicles to navigate complex environments safely.In this hybrid position, you will collaborate closely with our PrePlan team and be matched with your team after completing onsite interviews.Your Responsibilities Will Include:Developing cutting-edge ML-driven prediction and planning systems to boost the capabilities of our ML driver and facilitate the rapid growth of Waymo’s services.Converting open-ended, real-life challenges into well-defined ML problems, researching and applying state-of-the-art ML techniques, such as foundation models and reinforcement learning, to enhance autonomous vehicle planning and prediction.Collaborating with top-tier researchers, engineers, and product managers to ensure safe and smooth planning behaviors for all road users, aligning with product specifications.Creating and assessing large-scale models, integrating them into Waymo’s production planning software for real-world applications in close cooperation with the Planner and Research teams.

Who We Are Nuro is a pioneering self-driving technology company dedicated to making autonomy accessible to everyone. Established in 2016, we are developing the most scalable autonomous driving solution, merging advanced AI with high-performance automotive hardware. Our flagship technology, the Nuro Driver™, is licensed for various applications including robotaxis, commercial fleets, and personal vehicles. With years of proven self-driving deployments, Nuro provides automakers and mobility platforms with a clear trajectory towards commercial-scale AVs, fostering a safer, more connected future. About the Role Join our learned behavior team where we leverage sophisticated machine learning strategies to expedite software advancements. This role involves close collaboration with software teams to identify challenges and implement innovative machine learning approaches to tackle real-world problems. Your work may include self-supervised learning for robust representation, out-of-distribution detection for addressing long-tail issues, refining reinforcement learning methods for motion planning, trajectory prediction, and enhancing model robustness to address uncertainties. If you are passionate about solving complex problems with practical solutions ready for vehicle deployment, we invite you to be a part of our journey! About the Work Develop scalable machine learning systems for planning and predicting safe, feasible trajectories for autonomous vehicles. Investigate generative sequence modeling and sequential decision-making; backgrounds in Embodied AI for robotics, Causal reasoning, Model interpretability, and Joint prediction and planning are advantageous. Address uncertainties arising from interconnected autonomous systems. Collaborate across autonomy teams to create comprehensive solutions for major autonomy challenges by understanding issues, proposing ideas, prioritizing tasks, and developing effective solutions. Implement practical solutions and deploy them on real-world vehicles. About You You possess extensive expertise and experience in one or more of the following areas: M.Sc. or Ph.D. in Computer Science, Artificial Intelligence, Mathematics, or a related discipline. Strong background in sequential decision-making, Imitation Learning, Deep Reinforcement Learning, generative modeling, and more.

Waymo is at the forefront of autonomous driving technology, striving to be the world's most trusted driver. Originating as the Google Self-Driving Car Project in 2009, Waymo has dedicated itself to developing the Waymo Driver—The World’s Most Experienced Driver™—to enhance mobility access and save countless lives that are lost to traffic accidents. The Waymo Driver powers our fully autonomous ride-hailing service and can be integrated into various vehicle platforms and applications. With over ten million rider-only trips completed and extensive experience gained from autonomously driving over 100 million miles on public roads and tens of billions of miles in simulation across more than 15 states, we are leading the way in transforming transportation.The Waymo Applied Research team is committed to pioneering machine learning solutions that tackle significant challenges in autonomous driving, with the ultimate aim of safely operating Waymo vehicles across numerous cities and diverse driving conditions. Our team actively initiates and nurtures collaborations with other research teams within Alphabet to drive innovation.In this hybrid role, you will report directly to a Technical Lead Manager.

About NuroNuro is pioneering self-driving technology with a vision to make autonomy universally accessible. Established in 2016, we are committed to developing the most scalable driving solution, integrating advanced AI with automotive-grade hardware. Our flagship technology, the Nuro Driver™, is licensed to facilitate various applications, including robotaxis, commercial fleets, and personal vehicles. With proven technology from numerous self-driving deployments, Nuro provides automakers and mobility platforms a clear pathway to achieving commercial-scale autonomous vehicles, fostering a safer, more connected future.Role OverviewAs a Lead Machine Learning Research Scientist, you will collaborate with our autonomy teams to address complex perception, prediction, and planning challenges using innovative machine learning techniques. This includes working with foundation models, multi-modal large language models, generative models, out-of-distribution detection, imitation learning, and reinforcement learning. If you thrive on tackling challenging problems and translating solutions into real-world applications, we invite you to join our team!Key ResponsibilitiesResearch and implement cutting-edge machine learning techniques.Address state-of-the-art perception, prediction, and planning issues within our autonomy system.Develop and optimize efficient ML data pipelines.Provide technical leadership and mentorship to a diverse team.Deploy autonomous software solutions on public roads.

Who We AreNuro is at the forefront of self-driving technology, dedicated to making autonomy accessible for everyone. Founded in 2016, we are developing the most scalable autonomous driving system, merging advanced AI with automotive-grade hardware. Our flagship technology, the Nuro Driver™, is licensed for various applications, including robotaxis, commercial fleets, and personal vehicles. With years of successful deployments, Nuro provides automakers and mobility platforms a clear pathway to commercial-scale autonomous vehicles, paving the way for a safer and more connected future.About the RoleThe Prediction team is tasked with leveraging cutting-edge machine learning techniques to enhance the functionality of the Nuro Driver. As a pivotal member of the Prediction and Smart Agents team, your role will center on developing state-of-the-art models that predict the behavior of surrounding traffic, which are integral to our autonomous systems. These models will be utilized both on-board and off-board in realistic closed-loop simulations.In this position, you will delve into innovative machine learning approaches to address complex challenges in autonomous driving. Your work will involve employing generative sequence modeling techniques to accurately predict intricate, interactive traffic scenarios. This necessitates a deep understanding of the intentions of other road users and their impact on safe driving decisions. You will also explore diverse input modalities, including End-to-End (E2E) strategies, to forecast the behavior of other agents. A crucial aspect of this role is the creation of intelligent, controllable agents to facilitate effective closed-loop training within simulations.If you are driven by the challenge of solving complex problems, leading impactful research, and witnessing your contributions in real-world applications, we invite you to apply!About the WorkDesign and develop scalable, machine learning-based prediction systems to generate multi-modal, realistic, and kinematically feasible trajectories.Engage in pioneering research focused on generative sequence modeling and sequential decision-making, exploring areas such as:Scalable generative sequence modeling techniques.Modeling marginal, conditional, and joint distributions for interactive agents.Transformer-based encoder-decoder frameworks.Large generative models and diffusion techniques.Agent controllability through conditioning, guidance, and various methodologies.Collaborate closely with the Planning team to create realistic and controllable agents that enhance our simulation capabilities.

Waymo is at the forefront of autonomous driving technology, dedicated to becoming the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, our mission centers on the Waymo Driver—The World’s Most Experienced Driver™—aimed at enhancing mobility access while significantly reducing traffic-related fatalities. The Waymo Driver currently underpins our fully autonomous ride-hailing service and is versatile enough to be integrated into various vehicle platforms and applications. With over ten million rider-only trips completed, our technology has autonomously navigated more than 100 million miles on public roads and executed tens of billions of miles in simulation across over 15 states in the U.S.The Waymo AI Foundations team is committed to advancing machine learning solutions that address the challenges of autonomous driving, enabling the safe operation of Waymo vehicles in numerous cities and under diverse driving conditions. We actively pursue collaborations with other research teams within Alphabet. Our current focus areas encompass reinforcement learning, learning from demonstration, generative modeling, Bayesian inference, hierarchical learning, and robust evaluation.In this hybrid role, you will report directly to a Principal Scientist.

About Us Nuro is a pioneering self-driving technology company striving to make autonomous vehicles accessible for everyone. Since our founding in 2016, we have been dedicated to creating the world’s most scalable driver, integrating advanced artificial intelligence with automotive-grade hardware. Our flagship technology, the Nuro Driver™, is licensed to support various applications, including robotaxis, commercial fleets, and personally owned vehicles. With a track record of successful self-driving deployments, Nuro is paving the way for automakers and mobility platforms to realize the commercial potential of autonomous vehicles, fostering a safer, more connected future. Role Overview As a Senior Machine Learning Research Scientist focusing on Vision-Language-Action (VLA) models, you will enhance our onboard Behavior & Planning stack to promote safe and robust decision-making in complex driving scenarios. Your expertise will drive the development of multimodal models that integrate scene understanding, contextual reasoning, and planning-relevant representations for real-world autonomous driving. This position emphasizes advancing cutting-edge VLA models, including model development, large-scale training, fine-tuning, evaluation, and optimization for onboard deployment. You will collaborate closely with teams across behavior, planning, perception, systems, and infrastructure to translate research breakthroughs into practical applications deployed in our vehicles. If you are passionate about creating and implementing state-of-the-art VLA systems in robotics, we encourage you to apply. Key Responsibilities Develop and enhance VLA models for onboard Behavior & Planning in autonomous driving systems. Create multimodal models that facilitate safe decision-making in complex and ambiguous driving situations. Research and implement state-of-the-art techniques in vision-language-action modeling, multimodal representation learning, and foundational models for autonomy. Train, fine-tune, and evaluate large-scale VLAs utilizing diverse and real-world driving datasets to improve model quality and robustness. Optimize models for efficient onboard deployment, focusing on inference speed, memory usage, and runtime performance. Collaborate with various teams to define training, evaluation, and deployment requirements. Design effective evaluation methodologies for multimodal models in safety-critical scenarios. Contribute to scalable model and data pipelines that support rapid experimentation and deployment. Qualifications Proven expertise in machine learning, particularly in vision-language-action frameworks. Experience with multimodal model development and evaluation. Strong background in autonomous driving systems and decision-making processes. Familiarity with large-scale data training and optimization techniques. Excellent collaboration skills to work effectively with cross-functional teams.

About Contextual AI At Contextual AI, we are at the forefront of transforming the functionality of AI Agents by addressing one of the most significant challenges in artificial intelligence: context. By providing the right context at the right moment, we enhance the precision and scalability that enterprises leveraging AI demand. Our comprehensive enterprise AI development platform bridges the gap between groundbreaking AI research and the practical needs of developers. It empowers AI developers to efficiently ingest and query documents from various enterprise data sources, seamlessly integrating retrieval results into their business processes. Founded by the trailblazers of Retrieval-Augmented Generation (RAG), the core technique that connects foundational models to real-time, relevant information, Contextual AI is supported by visionary venture capitalists. We are not merely participants in the enterprise AI revolution; we are shaping its direction. Join us in crafting a future where AI not only answers inquiries but also revolutionizes business operations. About the Role As a Research-focused Member of Technical Staff, you will engage in pioneering AI research initiatives, contributing significantly to the development of Contextual RAG pipelines, including retrieval systems, language models, and alignment methodologies. Responsibilities Lead research projects in language modeling, retrieval, alignment, retrieval-augmented generation, end-to-end training, and evaluation. Develop and implement scalable infrastructure and tools that facilitate effective research and model development. Continuously review and stay informed about current literature, advancements, and industry best practices. Participate in the entire research pipeline, from ideation and experimentation to analysis and deployment. Publish research findings that contribute novel insights to the field of AI. Qualifications Bachelor's degree in Computer Science, Software Engineering, or a related field; Master's or PhD preferred. Solid software engineering fundamentals with a proven history of developing complex systems. Deep understanding of core machine learning principles with hands-on experience in language modeling, retrieval, large-scale training, or evaluation. Research experience (academic or industry) is highly advantageous, including knowledge of experimental design, analysis, and research methodologies. Exceptional problem-solving capabilities and the ability to excel in a dynamic research environment.

About Nuro Nuro is at the forefront of self-driving technology, dedicated to making autonomy accessible to everyone. Established in 2016, we are developing the world’s most scalable autonomous driver, integrating state-of-the-art AI with high-quality automotive hardware. Our proprietary technology, the Nuro Driver™, is licensed for a diverse array of applications, including robotaxis, commercial fleets, and personal vehicles. Proven through extensive self-driving deployments, Nuro provides automakers and mobility platforms a clear pathway to commercial-scale autonomous vehicles, paving the way for a safer, more connected future. Role Overview As a Senior Machine Learning Research Scientist in our Perception & Behavior team, you will harness cutting-edge machine learning research to tackle complex real-world robotics challenges. Your primary focus will be on advancing the ML landscape and large-scale learning applications for autonomous vehicles, steering towards a holistic end-to-end driving system. This role involves developing and refining large-scale models for perception and behavior, staying abreast of the latest architectural innovations, collaborating with cross-functional teams to assess data and infrastructure needs, and enhancing model optimization and inference speeds. Your expertise in applied research will be critical in the creation and deployment of ML models within our autonomous vehicles. If you're passionate about solving fundamental AI challenges and seeing your solutions come to life in real-world robotics, we invite you to join our team! Key Responsibilities Develop and enhance state-of-the-art end-to-end ML models for autonomous vehicles, processing raw sensor data (camera, LiDAR, radar) to inform vehicle behavior. Rapidly prototype and iterate on solutions using extensive real-world driving datasets and robust computational resources. Investigate and implement innovative solutions to tackle some of the most pressing issues in AV technology, including handling rare events, ensuring safety, and providing comfortable user experiences. Collaborate closely with autonomy teams to identify critical challenges and devise comprehensive solutions through learned approaches. Partner with infrastructure, data, and evaluation teams to create efficient data handling, training, assessment, and validation processes. Deliver practical, high-performance solutions that can be deployed to the NuroDriver in live environments.

Waymo is at the forefront of autonomous driving technology, striving to become the world’s most reliable driver. Originating from the Google Self-Driving Car Project in 2009, Waymo has dedicated itself to developing the Waymo Driver—The World’s Most Experienced Driver™—to enhance mobility access and prevent countless lives lost to traffic accidents. Our cutting-edge Waymo Driver is the backbone of our fully autonomous ride-hailing service and can be implemented across various vehicle platforms and applications. Having completed over ten million rider-only trips, our technology has accumulated experience from autonomously driving over 100 million miles on public roads and tens of billions of miles in simulations across more than 15 U.S. states.The Large Model Evaluation team is pivotal to Waymo's AI mission. As advancements in Large Language Models (LLMs) and Vision-Language Models (VLMs) continue, we are constructing state-of-the-art AI systems capable of addressing the full spectrum of real-world driving complexities. Central to our success is our capacity to measure progress effectively. Robust evaluation is crucial for deploying any large model, and at Waymo, the challenge is particularly intricate and safety-critical. We seek quantitatively-minded engineers to innovate and propose new methods for assessing the machine learning models integrated into the Waymo Driver.

Waymo is at the forefront of autonomous driving technology, striving to become the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, Waymo is dedicated to advancing the Waymo Driver—The World's Most Experienced Driver™—to enhance mobility and save lives that are often lost to traffic accidents. The Waymo Driver not only powers our fully autonomous ride-hail service but is also adaptable across various vehicle platforms and use cases. With over ten million rider-only trips completed, our technology has driven autonomously for more than 100 million miles on public roads, supported by tens of billions of miles in simulation across 15+ U.S. states.Our Compute Team plays a pivotal role in delivering the compute platform that runs the software stack of fully autonomous vehicles. We specialize in designing high-performance custom silicon, developing cutting-edge system-level compute architectures that maximize performance, power efficiency, and minimize latency. Our multidisciplinary team thrives on collaboration, working closely with various engineering teams to ensure both hardware and software are optimized for peak performance. We are looking for passionate and talented individuals to join us in developing one of the highest-performing automotive compute platforms in the world.In this hybrid role, you will report to a Hardware Engineering Manager.

Waymo is a pioneering company in autonomous driving technology, dedicated to becoming the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, Waymo has been on a mission to develop the Waymo Driver—The World’s Most Experienced Driver™—which aims to enhance mobility access and prevent the thousands of fatalities caused by traffic accidents. The Waymo Driver is the backbone of our fully autonomous ride-hail service and is adaptable to various vehicle platforms and applications. With more than ten million rider-only trips completed, our technology has successfully navigated over 100 million miles on public roads and simulated tens of billions of miles across more than 15 U.S. states.The Perception team is responsible for creating the systems that learn the spatial-temporal representation and semantic meanings of the environment surrounding the autonomous driving vehicle (ADV). Our work involves close collaboration with downstream teams to optimize and integrate these systems into the Waymo Driver. We actively conduct research to solve real-world challenges and partner with research teams at Alphabet. With access to millions of driving data miles collected from a variety of sensors, our engineers have the opportunity to (1) devise methods for continuous learning from large-scale real-world data, (2) build and train models at scale, (3) analyze real-world behavior and develop systems to manage complex interactions with the environment, and (4) refine models for our onboard and offboard hardware.In this hybrid role, you will report to a Technical Lead Manager.

Waymo, an innovator in autonomous driving technology, is dedicated to becoming the world's most reliable driver. Originating as the Google Self-Driving Car Project in 2009, our focus has been on creating the Waymo Driver—The World’s Most Experienced Driver™—to enhance mobility access and prevent traffic-related fatalities. The Waymo Driver supports our fully autonomous ride-hail service and can be integrated into various vehicle platforms for diverse applications. With over ten million rides provided and experience from driving more than 100 million miles on public roads combined with billions of miles in simulation across 15+ U.S. states, we’re at the forefront of a transportation revolution.The Perception team is responsible for developing systems that learn spatial-temporal representations and their semantic meanings within the autonomous vehicle's environment. Collaborating closely with other teams for optimization and integration into the Waymo Driver, we engage in research to tackle practical challenges and work with Alphabet's research teams. With access to millions of miles of diverse driving data, our engineers can (1) create methods for efficient and continuous learning from large-scale real-world data, (2) build and train models at scale, (3) analyze real-world behavior to manage complex interactions, and (4) optimize models for our onboard and offboard hardware.In this hybrid role, you will report directly to a Technical Lead Manager.

Waymo is a pioneering force in autonomous driving technology, dedicated to becoming the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, Waymo's mission focuses on creating the Waymo Driver—The World’s Most Experienced Driver™—with the goal of enhancing mobility access and significantly reducing traffic-related fatalities. The Waymo Driver is the backbone of our fully autonomous ride-hailing service and is adaptable across various vehicle platforms and applications. With over ten million rider-only trips completed and the experience of driving autonomously for over 100 million miles on public roads, alongside tens of billions of miles in simulation across more than 15 U.S. states, we are committed to transforming the future of transportation.The Simulator Team at Waymo is at the forefront of innovation, developing advanced simulations that replicate realistic environments for the testing, training, and validation of the Waymo Driver. Our team comprises a diverse, collaborative group of machine learning (ML) engineers, software engineers, and ML research engineers, who are dedicated to crafting industry-leading simulation solutions. Utilizing cutting-edge generative and reconstructive ML algorithms, we model the real world, including realistic agents, road systems, traffic dynamics, weather conditions, and a comprehensive sensor suite (Camera, Lidar, Radar).In this exciting role, you will be instrumental in enhancing the fidelity, scalability, controllability, and richness of our simulations by exploring the latest advancements in 3D world modeling. By leveraging state-of-the-art ML technologies trained on extensive datasets, you will help us create dynamic, semantically rich virtual worlds that have a direct impact on the development and validation of the Waymo Driver.You will report directly to a Senior Staff Engineering Manager.Responsibilities:Lead the design, development, and deployment of innovative 4D world models and generative systems aimed at generating ultra-realistic and controllable sensor outputs and semantics for simulation applications at Waymo.Architect and implement scalable, robust ML pipelines for the training, evaluation, and deployment of large-scale generative models within our simulation framework, employing techniques like model distillation and quantization.Develop and scale production-ready video generation methodologies (e.g., Diffusion, Flow Matching) to create dynamic and interactive simulation environments.Utilize Vision Language Models (VLMs) to enhance semantic comprehension and control within our world simulation products.Collaborate with leading research teams across Waymo and Alphabet to integrate state-of-the-art research in 4D world modeling and generative AI into scalable, production-ready solutions.Provide mentorship and technical guidance to fellow engineers on the team.