2024 Intern, Software Engineer, ML Runtime Optimization

Key Skills

AI Robotic systems, C++, Computer Vision (CV), Deep Learning, JAX framework, Machine learning techniques, Natural Language Understanding-NLU, Python Programming, PyTorch, TensorFlow

Job Description

At Waymo, we are mission-driven and believe deeply in the opportunity of autonomous driving technology to improve mobility and make people's lives better. We are united by purpose and responsibility (for our employees and riders alike). We are looking for kind, committed, employees who have integrity, dream big, work together as one team and create a sense of belonging for one another that is the foundation of our culture. We want each team member to feel welcomed and included in every step of our exciting journey. 

In this internship, you will

• Define, plan and execute cutting-edge deep learning research to advance experiences
• Develop novel deep learning techniques to achieve state-of-the-art accuracy within the constraints of on-device and real-time execution.
• Collaborate with other research scientists and software engineers to develop innovative deep learning techniques for vision and robotics.
• Collaborate with software and hardware engineers to develop tradeoff curves for accuracy vs the runtime resources/constraints such as latency.
• Communicate the experimental results and the recommendations clearly, both within the group as well as to the cross-functional groups.

Qualifications

Waymo interns work alongside leaders in the industry on projects that deliver significant impact to the company. We believe learning is a two-way street: leveraging your knowledge while providing you with opportunities to expand your skill-set. Interns are an important part of our culture and our recruiting pipeline. Join us at Waymo for a fun and rewarding internship!

Required qualifications: 

• PhD candidate in Computer Science, EE, Mathematics or similar technical field of study
• Experience programming in Python, C++.
• Experience with developing deep learning algorithms on a ML framework, such as Pytorch, Tensorflow or JAX.
• Interests and a good understanding of HW system and model optimization techniques, such as quantization, distillation, low-rank approximation or efficient transformer. 

Preferred qualifications: 

• Publication track record in machine learning conferences and/or journals on the topics of efficient deep learning or ML System.
• Experience in optimizing deep learning networks for execution constraints besides just the final accuracy.
• Familiar with HW performance analysis. (or ML analysis system)
• Background in computer vision, natural language understanding, robotics.

The expected base salary for this full-time position is listed below. The salary will be prorated based on the number of weeks you are employed as an intern. Interns are also eligible to participate in the Company’s generous benefits programs, subject to eligibility requirements.

PhD Pay

$125,000—$125,000 USD

While at Waymo, you will enjoy benefits that cover…

Health and wellness: Our people are at the heart of everything we do. At Waymo, you can enjoy top-notch medical, dental and vision insurance, mental wellness support, a Flexible Spending Account (FSA), a Health Saving Account (HSA), on-site physicians and/or nurses in some locations, and special wellness programs.