Machine Learning SWE Infra

Who you are

You have a strong background in machine learning and software engineering, with at least 5 years of experience in developing and deploying machine learning models. Your expertise in Python and frameworks like TensorFlow and Keras allows you to build scalable systems that can handle complex data workflows. You are familiar with cloud platforms such as AWS and have experience with containerization tools like Docker, which you use to streamline deployment processes. Your passion for autonomous driving technology drives you to continuously improve the performance of machine learning systems, ensuring they meet the high standards required for safety and reliability.

You possess a deep understanding of large-scale machine learning model architectures, particularly in the context of autonomous vehicles. You are skilled in evaluating interesting driving behaviors and can design experiments that yield valuable insights into model performance. Your analytical mindset enables you to combine expert human judgments with advanced machine learning techniques, delivering training and evaluation data that enhances the Waymo Driver's capabilities.

You thrive in collaborative environments and enjoy working with cross-functional teams, including researchers and software engineers. Your ability to communicate complex technical concepts clearly helps you build strong relationships with stakeholders and contribute to the overall success of the team. You are committed to fostering a culture of innovation and continuous learning, encouraging others to share their ideas and insights.

Desirable

Experience with simulation workflows and insight tools is a plus, as it allows you to contribute to the development of systems that improve the evaluation and onboarding processes for developers. Familiarity with autonomous vehicle technologies and the challenges they present will further enhance your ability to make impactful contributions to the team.

What you'll do

As a Senior Machine Learning Engineer at Waymo, you will lead the development of scalable machine learning systems that support the evaluation of the Waymo Driver. You will design and implement large-scale models that analyze driving behaviors, ensuring that the technology meets the highest safety standards. Your role will involve collaborating with researchers to integrate advanced machine learning techniques into the evaluation systems, enhancing the overall performance of the Waymo Driver.

You will be responsible for building and optimizing data pipelines that facilitate the training and fine-tuning of machine learning models. Your expertise in AWS will be crucial as you leverage cloud resources to scale your solutions effectively. You will also work closely with software engineers to ensure that the systems you develop are robust, efficient, and easy to maintain.

In addition to technical responsibilities, you will mentor junior engineers and contribute to the team's knowledge-sharing culture. You will participate in code reviews and provide constructive feedback to help your colleagues grow in their roles. Your leadership will be instrumental in driving the team's success and fostering an environment of collaboration and innovation.

What we offer

Waymo offers a competitive salary range of $238,000—$302,000 USD, along with eligibility for an annual bonus program and equity incentive plan. You will have access to a generous benefits program that supports your well-being and professional growth. Join us in our mission to revolutionize transportation and make the world a safer place through autonomous driving technology.