Machine Learning Engineer (Perception)

• Our goal is to improve the Waymo Driver performance by owning the end-to-end driving behavior for various onroad problems clusters
• We work with upstream and the downstream teams on developing the ML models and integrating into the Waymo Driver
• We have access to millions of miles of driving data from a diverse set of sensors, enabling engineers like you to (1) develop methods for efficiently and continuously learning from large scale real-world data, to (2) develop models and model training at scale, to (3) analyze real-world behavior and develop systems for handling the complexities of interacting with the real-world, and (4) optimize models for our onboard and offboard hardware
• In this hybrid role, you will report to a Technical Lead Manager
• Own tasks in the ML Driver, take responsibility for task performance, create ML methods and recipes to scale and improve tasks
• Analyze behavior of ML systems in real-world application, identify issues and root causes, advise or develop short- and long-term solutions
• Develop methods and recipes for distributed fine-tuning enabling multiple developers to simultaneously improve the model, develop methods and recipes to avoid regression against a production system
• Monitor ML systems in production, develop methods for automatically detecting issues or regressions, develop AI-aided analysis and debugging tooling
• Develop and maintain model evaluation recipes and metrics for measuring and improving performance of pre-trained and fine-tuned models

Benefits

• Medical, dental, and vision insurance for employees and dependents
• Employee assistance programs focused on mental health
• Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations
• Access to mental health apps
• Onsite wellness centers
• Counseling services
• Second medical opinion for you and your loved ones
• Medical advocacy program for transgender employees
• Support programs including menopause benefit
• Competitive compensation
• Regular bonus and equity performance grant opportunities
• Generous 401(k) and regional retirement plans
• Annual cross-company compensation review and pay equity analysis
• 1-on-1 financial coaching
• Fertility and growing family assistance
• Parental leave and baby bonding leave
• Elder care and support
• Survivor income benefit
• Backup childcare
• Caregiver leave
• Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays
• Jury duty leave
• Military leave
• Hybrid work model with remote work opportunities also available
• Educational reimbursement
• Peer learning and coaching platform
• Donation matching programs
• Employee resource groups
• Volunteer hours
• Internal community groups and local culture clubs
• Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts
• On-site meals and snacks
• Fitness centers, massage programs, and ergonomic support
• On-demand fitness, wellbeing, and cooking classes
• Commuter benefits- Experience with Python or C++
• Experience with ML frameworks like PyTorch or JAX
• Bachelors in Computer Science or a similar discipline, or an equivalent amount of deep learning experience
• 3+ years experience in Machine Learning and/or Computer Vision
• Publications at top-tier conferences like CVPR, ICCV, ECCV, ICLR, ICML, ICRA, IROS, RSS, NeurIPS, AAAI, IJCV, PAMI
• MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline