Revolutionizing transportation with autonomous driving
Waymo, a subsidiary of Alphabet Inc., is at the forefront of autonomous driving technology, operating robotaxis in cities like Phoenix, San Francisco, and Los Angeles. With over 10 million fully driverless rides and more than 100 million miles driven, Waymo is transforming transportation. The compan...
Waymo offers comprehensive medical, dental, and vision insurance for employees and their dependents, along with commuter benefits and onsite wellness ...
Waymo fosters a culture of innovation and safety, focusing on the real-world application of autonomous technology. The company values diversity and in...
Waymo • Mountain View, CA, USA; San Francisco, CA, USA; Bellevue, WA, USA
Waymo is hiring a Staff Machine Learning Engineer focused on ML Performance & Optimization. You'll optimize neural model architectures for high performance across multiple platforms, utilizing skills in TensorFlow, Python, and Keras. This role requires extensive experience in machine learning and model optimization.
You have a strong background in machine learning with at least 5 years of experience in optimizing neural networks and model architectures. Your expertise in TensorFlow and PyTorch allows you to implement cutting-edge solutions that enhance model performance across various platforms. You are proficient in Python and have hands-on experience with GPU and TPU systems, enabling you to tackle complex challenges in model training and inference.
You stay updated with the latest advancements in machine learning and are eager to apply innovative techniques to improve production systems. Your collaborative spirit shines through as you work closely with cross-functional teams, including ML Engineering and Modeling, to drive the scaling of models and ensure their effectiveness across multiple applications. You possess excellent problem-solving skills and a keen analytical mindset, allowing you to dissect intricate issues and propose effective solutions.
Experience with large-scale data processing and optimization techniques is a plus. Familiarity with cloud platforms and distributed computing will further enhance your ability to contribute to Waymo's mission. You are passionate about autonomous driving technology and are motivated by the impact your work can have on improving mobility and safety.
In this role, you will be responsible for optimizing neural model architectures and systems to achieve high performance on both GPU and TPU platforms. You will collaborate with the ML Engineering and Modeling team to ensure that the models you develop can scale effectively and support multiple platforms. Your work will involve analyzing model performance, identifying bottlenecks, and implementing solutions that enhance efficiency and reliability.
You will also engage in research to explore the latest developments in efficient machine learning, integrating these innovations into Waymo's production systems. Your contributions will directly impact the performance of the Waymo Driver, which powers the company's autonomous ride-hail service. You will be expected to document your findings and share insights with the team, fostering a culture of continuous improvement and knowledge sharing.
Waymo provides 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 generous company benefits, supporting your well-being and professional growth. As part of a pioneering team in autonomous driving technology, you will have the opportunity to make a significant impact on the future of mobility. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds.
Apply now or save it for later. Get alerts for similar jobs at Waymo.
