Staff Tech Lead (Machine Learning Engineer, Perception)

• The Perception team builds the system which learns the spatial-temporal representation and their semantic meanings of the surrounding environment of the self-driving car, i.e., the system that “perceives” the world around the car
• We work jointly with downstream teams on the optimization and integration into the Waymo Driver
• We conduct our own research to address real-world problems and collaborate with research teams at Alphabet
• We have access to millions of miles of driving data from a diverse set of sensors, enabling researchers like you to develop models for panoptic segmentation and occupancy in 3D and over time
• In this hybrid role you will report to a Sr Staff Technical Lead Manager
• Own the panoptic segmentation task (instance segmentation and semantic segmentation), take responsibility for task scaling and task performance, take ownership of aligning instance segmentation and object detection/tracking (in particular in an end-to-end architecture), create ML methods and recipes to scale and improve tasks
• Collaborate with ML engineering/research teams to develop efficient model architectures for representing occupancy and panoptic segmentation, and train and deploy such models with a focus on using large pre-trained models
• Collaborate with partner teams to establish efficient human and machine labeling workflows
• Analyze panoptic segmentation issues in real-world application, identify issues and root causes, advise or develop short- and long-term solutions
• Develop and maintain metrics for 4D (3D + time) instance segmentation and semantic segmentation

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- Bachelors in Computer Science or a similar discipline, or an equivalent amount of deep learning experience
• Experience with ML frameworks like PyTorch or JAX
• 7+ years experience in Machine Learning and/or Computer Vision
• Experience with Python
• MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline
• Publications at top-tier conferences like CVPR, ICCV, ECCV, ICLR, ICML, ICRA, IROS, RSS, NeurIPS, AAAI, IJCV, PAMI
• Experience with C++