As a Machine Learning Engineer, GPU Kernel and Runtime at Waymo, you are contributing to the development of the world's most trusted driver, revolutionizing the future of autonomous driving. This role entails designing, developing, and optimizing GPU kernels and runtime systems to accelerate machine learning workloads, which is crucial for the success of Waymo's autonomous driving technology. By joining our team, you are becoming part of a mission to improve access to mobility while saving thousands of lives lost to traffic crashes. The impact and scope of this position are significant, as you will be working on cutting-edge technology that has the potential to transform the entire transportation industry. With Waymo's commitment to innovation and growth, this role offers tremendous opportunities for professional advancement and development, allowing you to expand your skills and expertise in machine learning, GPU computing, and autonomous driving.

As a Machine Learning Engineer, GPU Kernel and Runtime, your key responsibilities include:

• Designing and developing high-performance GPU kernels for machine learning algorithms, focusing on optimization, scalability, and reliability


• Collaborating with cross-functional teams to integrate GPU kernels with machine learning frameworks and runtime systems


• Developing and maintaining tools and scripts for kernel development, testing, and deployment


• Investigating and resolving performance bottlenecks and other technical issues in GPU kernel and runtime systems


• Participating in code reviews and contributing to the improvement of the overall code quality and architecture


• Staying up-to-date with the latest advancements in GPU computing, machine learning, and autonomous driving, applying this knowledge to drive innovation and improvement


• Working closely with the research team to develop and implement new machine learning algorithms and techniques. You will be reporting to the technical lead and working closely with a team of experienced engineers, researchers, and technicians. To be successful in this role, you need to possess strong technical skills in GPU computing, machine learning, and software development, as well as excellent communication and collaboration skills.

By joining Waymo as a Machine Learning Engineer, GPU Kernel and