Machine Learning Engineer, Runtime & Optimization

📋 Description

• Lead collaboration with Waymo ML scientists in perception, planner, research and simulation.
• Lead projects from proposals through execution by developing junior engineers.
• Analyze and improve ML system workloads on both cloud and self-driving cars.
• Apply model optimization, efficient deep learning techniques and ML software improvements to
• Work across the ML stack from efficient models, model compression, to ML software (e.g., JAX, XLA

🎯 Requirements

• M.S. in CS, EE, Deep Learning or a related field
• 2+ years of experience as a technical lead, including writing project plans, mentoring, and
• 5+ years of experience developing ML systems or ML software stack (PyTorch/JAX/TF, runtime
• Deep understanding of ML system architecture, performance analysis and tools.
• Strong Python or C++ programming skills

🎁 Benefits

• Discretionary annual bonus program
• Equity incentive plan
• Generous company benefits program