Staff ML Engineer (Foundation Models)

• The mission of the Waymo AI Foundations team is to develop machine learning solutions addressing open problems in autonomous driving, towards the goal of safely operating Waymo vehicles in dozens of cities and under all driving conditions
• As part of our work, we also initiate and foster collaborations with other research teams in Alphabet. AI Foundations areas that we are currently focusing on include reinforcement learning, learning from demonstration, generative modeling, Bayesian inference, hierarchical learning, and robust evaluation
• This role follows a hybrid work schedule and reports to a Director of AI Foundations
• Work on connecting large-scale foundation models with production systems
• Adapt our foundation models to new sensors, new platforms, and new production requirements
• Collaborate extensively with other teams to land foundation models to our next-gen platforms

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
• Second medical opinion for you and your loved ones
• Counseling services
• Support programs including menopause benefit
• Medical advocacy program for transgender employees
• Competitive compensation
• Regular bonus and equity performance grant opportunities
• Generous 401(k) and regional retirement plans
• 1-on-1 financial coaching
• Annual cross-company compensation review and pay equity analysis
• Fertility and growing family assistance
• Parental leave and baby bonding leave
• Survivor income benefit
• Elder care and support
• 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- Track record of solving complex system problems
• Experience in building large-scale systems
• Track record of successful deliveries through large-scale collaborations
• Experience in LLM/VLM related foundation models or systems
• Experience with foundation model post-training