Senior Software Engineer, Training Efficiency

The Waymo ML Infrastructure team works with Research and Production teams to develop models in Perception and Planning that are core to our autonomous driving software. We help our partners by offering the best solutions for the entire model development lifecycle. These solutions are developed in close collaboration with teams at Google. They are geared towards both scaling models and solving problems unique to ML for autonomous driving. You will improve the runtime efficiency of input data pipelines for large-scale training workloads. This is a unique opportunity to work on ML systems and improve on our model training processes.