Machine Learning Engineer, Runtime & Optimization
Machine Learning Engineer, Runtime & Optimization
Machine Learning Engineer, Runtime & Optimization
Waymo
Waymo, an autonomous driving technology company, is seeking a Machine Learning Engineer for its Runtime & Optimization team. The role involves improving compute performance for machine learning workflows on both cloud and self-driving cars, collaborating with ML scientists, and leading projects to optimize ML systems.
Qualification
- M.S. in Computer Science, Electrical Engineering, Deep Learning, or a related field.
- 2+ years of experience as a technical lead, including project planning and mentoring.
- 5+ years of experience in developing solutions in ML systems or ML software stack (Pytorch/JAX/TF, runtime libraries, ML compiler).
- Deep understanding of ML system architecture, performance analysis, and tools.
- Strong programming skills in Python or C++.
- Experience with model optimization and efficient deep learning techniques.
Responsibility
- Lead collaboration with Waymo ML scientists in perception, planner, research, and simulation to enhance ML workloads.
- Manage projects from proposals through execution, mentoring junior engineers.
- Analyze and improve ML system workloads on cloud and self-driving cars.
- Apply model optimization and efficient deep learning techniques to Waymo's ML systems.
- Develop and implement improvements in ML software.
