Senior/Staff ML Engineer, 3D/4D World Modeling, Simulation

Who you are

You have 5+ years of experience in machine learning engineering, particularly in developing simulation solutions that model complex environments. Your background includes working with generative and reconstructive ML algorithms, and you are adept at creating dynamic virtual worlds that enhance the training and validation of autonomous systems.

You possess strong programming skills in Python and are familiar with frameworks such as TensorFlow and Keras. Your experience includes leveraging large-scale datasets to improve the fidelity and scalability of simulations, and you understand the intricacies of modeling realistic agents, traffic systems, and sensor data.

You thrive in collaborative environments and enjoy working with diverse teams of engineers and researchers. Your communication skills allow you to articulate complex technical concepts to both technical and non-technical stakeholders, fostering a culture of innovation and teamwork.

Desirable

Experience with 3D modeling tools and simulation software is a plus. Familiarity with autonomous driving technologies and the challenges associated with real-world applications will set you apart. You are passionate about pushing the boundaries of machine learning and simulation technologies to create impactful solutions.

What you'll do

In this role, you will be responsible for developing state-of-the-art simulations that test and validate the Waymo Driver. You will collaborate with cross-functional teams to design and implement advanced algorithms that model realistic environments, including agents, roads, and traffic systems. Your work will directly influence the development of autonomous driving technologies, ensuring safety and reliability.

You will leverage your expertise in machine learning to enhance the scalability and controllability of simulations, pushing the frontiers of 3D world modeling. Your contributions will involve experimenting with various ML techniques and optimizing algorithms for performance and accuracy.

You will also participate in code reviews and contribute to the continuous improvement of the team's processes and methodologies. By mentoring junior engineers and sharing your knowledge, you will help cultivate a culture of learning and growth within the team.

What we offer

Waymo provides a competitive salary range of $204,000—$259,000 USD, along with eligibility for an annual bonus program and equity incentive plan. You will have access to generous company benefits, including health and wellness programs, professional development opportunities, and a collaborative work environment that values diversity and innovation.

Join Waymo and be part of a mission-driven team that is transforming mobility and making roads safer for everyone. Your work will have a direct impact on the future of transportation, and you will be at the forefront of technological advancements in autonomous driving.